【pytorch】预训练模型的使用

pytorch自带有一些高级的复杂模型，我们可以通过 torchvision.models 调用，例如torchvision.models.densenet169(pretrained=True)就调用了densenet169的预训练模型。

先来看看这个模块涵盖了哪些模型：
见https://pytorch.org/vision/stable/models.html获取详细信息。

注意

基本所有预训练模型都是基于ImageNet数据集训练的，得先从从网站（https://s3.amazonaws.com/deep-learning-models/image-models/imagenet_class_index.json）下载一个json文件，然后通过以下代码提取0-999每个数字对应的文本类标签：

# ImageNet class index to label 
## ref: https://discuss.pytorch.org/t/imagenet-classes/4923/2
idx_to_label = json.load(open