PyTorch在哪些ImageNet类上进行训练？

Question

PyTorch允许您在此处下载预先训练的模型进行分类，但尚不清楚如何从概率的输出张量回到实际的类标签。有什么想法吗？

Answer 1

请参阅此处的示例：https://discuss.pytorch.org/t/imagenet-classes/4923 https://www.learnopencv.com/pytorch-for-beginners-image-classification-using-pre-trained-models/

您需要下载imagenet_classes.txt