facebookresearch / WSL-Images

Weakly Supervised Learning On Images
Other
597 stars 63 forks source link

Pretrained hashtag prediction model #13

Open sylvainma opened 4 years ago

sylvainma commented 4 years ago

Hello,

These models are the models ultimately used for Imagenet classification. I am looking for the pre-trained models used to predict hashtags. From the paper:

"In our experiments, we pre-train convolutional networks for hashtag prediction [...]" "Full network finetuning is performed by removing the hashtag-specific fully connected classification layer from the network"

I am looking for this hashtag prediction model for an open-source project which needs to clean and improve (by adding more) hashtags from Flickr.

At least it would help a lot knowing which layers of the available model on pytorch hub correspond to this underlying hashtag prediction model.

Thanks!

diegovalenzuelaiturra commented 4 years ago

Hi, have you checked this ?

sylvainma commented 4 years ago

Thanks Diego! I had a look but I think the problem is the same, they don't provide the model for hashtag prediction used as a backbone (and they don't specify which layer in the final model corresponds to it). I guess this issue could be asked there too.

vyaslkv commented 4 years ago

@sylvainma did you get anything for hashtag prediction

sylvainma commented 4 years ago

@vyaslkv not yet