PyTorch позволяет загружать предварительно обученные модели для классификации здесь , но не ясно, как go из выходного тензора вероятностей вернуться к фактическим меткам классов. Любые идеи?
Смотрите пример здесь: https://discuss.pytorch.org/t/imagenet-classes/4923 https://www.learnopencv.com/pytorch-for-beginners-image-classification-using-pre-trained-models/
Вам необходимо скачать imagenet_classes.txt
imagenet_classes.txt