ruotianluo / ImageCaptioning.pytorch

I decide to sync up this repo and self-critical.pytorch. (The old master is in old master branch for archive)
MIT License
1.43k stars 409 forks source link

Can I extract features using other pertained models except resnet? #185

Open qq123aa456 opened 10 months ago

qq123aa456 commented 10 months ago

except resnet,any other pertrained models can be used to extract features by prepo-feats.py?

ruotianluo commented 10 months ago

You would need to modify the code to do that. Shouldn't be too hard though.

qq123aa456 commented 10 months ago

I successful got it,thank you so much!