Obtenir le label ImageNet pour un index spécifique dans le tenseur de sortie 1000-dimensionnel dans la torche

Question

J'ai le Tensor de sortie d'une passe avant pour a Facebook implementation of the ResNet model avec une image chat. C'est un Tenseur de 1000 dimensions avec les probabilités de classification. En utilisant torch.topk je peux obtenir les probabilités du top 5 et leurs index dans le tenseur de sortie. Maintenant, je veux voir les étiquettes lisibles par l'homme pour ces index les plus probables.

J'ai cherché en ligne la liste des étiquettes (qui sont apparemment aussi appelé sysnets) et seulement trouvé ceci: http://image-net.org/challenges/LSVRC/2015/browse-synsets

Je mets ces étiquettes dans un fichier en utilisant les numéros de ligne comme l'indice d'étiquette et quand je lance la réseau avec deux images de chat différentes, je reçois "tournevis" comme le top deviner pour les deux. Si je trier le fichier d'étiquettes par ordre alphabétique, j'obtiens le terme "cinéma" pour les deux.

Cela semble être un problème avec la conversion d'index en étiquette, non? Alors ... la question est: Comment est-ce que je peux mapper correctement l'index dans le tenseur de sortie réseau à l'étiquette Imagenet?

Answer 1

Trouvé this tutorial on training ConvNets on ImageNet by Dato et à la fin il contient le mappage correct. Compte-rendu ici pour mémoire:

{ 
0: 'tench, Tinca tinca', 
1: 'goldfish, Carassius auratus', 
2: 'great white shark, white shark, man-eater, man-eating shark, Carcharodon carcharias', 
3: 'tiger shark, Galeocerdo cuvieri', 
4: 'hammerhead, hammerhead shark', 
5: 'electric ray, crampfish, numbfish, torpedo', 
6: 'stingray', 
7: 'cock', 
8: 'hen', 
9: 'ostrich, Struthio camelus', 
10: 'brambling, Fringilla montifringilla', 
... [truncated for space] 
990: 'buckeye, horse chestnut, conker', 
991: 'coral fungus', 
992: 'agaric', 
993: 'gyromitra', 
994: 'stinkhorn, carrion fungus', 
995: 'earthstar', 
996: 'hen-of-the-woods, hen of the woods, Polyporus frondosus, Grifola frondosa', 
997: 'bolete', 
998: 'ear, spike, capitulum', 
999: 'toilet tissue, toilet paper, bathroom tissue' 
} 

cartographie complète ici: https://gist.github.com/maraoz/388eddec39d60c6d52d4