ps:fendo为我自己的用户名
如何从pytorch官网下载一些常用的模型呢操作如下(选择你需要的模型即可)
import torchvision.models as modelsresnet18 = models.resnet18()alexnet = models.alexnet()vgg16 = models.vgg16()squeezenet = models.squeezenet1_0()densenet = models.densenet161()inception = models.inception_v3()googlenet = models.googlenet()shufflenet = models.shufflenet_v2_x1_0()mobilenet = models.mobilenet_v2()resnext50_32x4d = models.resnext50_32x4d()wide_resnet50_2 = models.wide_resnet50_2()mnasnet = models.mnasnet1_0()# 如果需要下载预训练好的模型,添加pretrained=True即可resnet18 = models.resnet18(pretrained=True)alexnet = models.alexnet(pretrained=True)squeezenet = models.squeezenet1_0(pretrained=True)vgg16 = models.vgg16(pretrained=True)densenet = models.densenet161(pretrained=True)inception = models.inception_v3(pretrained=True)googlenet = models.googlenet(pretrained=True)shufflenet = models.shufflenet_v2_x1_0(pretrained=True)mobilenet = models.mobilenet_v2(pretrained=True)resnext50_32x4d = models.resnext50_32x4d(pretrained=True)wide_resnet50_2 = models.wide_resnet50_2(pretrained=True)mnasnet = models.mnasnet1_0(pretrained=True)