xingtianyao的博客

import torchimport torch.nn as nnfrom torch.utils.data import dataset,dataloader,Dataset,DataLoaderfrom torchvision import transformsfrom torchvision.datasets import ImageFolderfrom torchvision.models import vgg16vgg = vgg16(pretrained=True)#固定...