search

Alibaba-MIIL / ImageNet21K

Official Pytorch Implementation of: "ImageNet-21K Pretraining for the Masses"(NeurIPS, 2021) paper
MIT License
732 stars 71 forks source link

ImageNet 21K-P Class ID:Name mapping #12

Closed ArikVoronovRazor closed 3 years ago

ArikVoronovRazor commented 3 years ago

Hi!

Thanks for the great repo!

I'm using timm to load your pretrained models. The "_in21k" variants (e.g tresnet_m_miil_in21k) output 11221 classes in the final layer, which seems to be the amount of classes in the Fall 11 Version.

Where can I find the mapping from the model output to the class names/wordnet IDs?

mrT23 commented 3 years ago

see an example in https://github.com/Alibaba-MIIL/ImageNet21K/blob/main/visualize_detector.py (line 64)