Getting weights of attribute head from Caffe model

[zhaohengz]

Hi,

Thanks for sharing the weights converted from the Caffe model. But it seems that the weights of the attribute head are removed. Is there way to convert the weights of the attribute head from Caffe model also?

Really appreciate your advice.

Thanks!

[airsplay]

Thanks.

The model & weight & config with attributes have been uploaded. Please check the demo here: https://github.com/airsplay/py-bottom-up-attention/blob/master/demo/demo_feature_extraction_attr.ipynb

[yezhengli-Mr9]

Hi,

Thanks for sharing the weights converted from the Caffe model. But it seems that the weights of the attribute head are removed. Is there way to convert the weights of the attribute head from Caffe model also?

Really appreciate your advice.

Thanks!

Hi @zhaohengz , I comprehend your question as how to convert weights from caffe to pytorch here. If so, I am also interested in this since I am to use a smaller backbone (for example, resnet18 instead of resnet101).