获取torchvision的预训练网络的分类标签

import torchvision.models as models
import json

model = models.resnet50(pretrained=True)

# 下载ImageNet的标签文件
!wget https://raw.githubusercontent.com/anishathalye/imagenet-simple-labels/master/imagenet-simple-labels.json

# 读取标签文件
with open('imagenet-simple-labels.json') as f:
    labels = json.load(f)

# 打印分类标签
print(labels)