pytorch vgg19 加载预训练模型做识别
import torchimport torchvisionimport torchvision.transforms as transformsfrom PIL import Imageimport numpy as nptrans = transforms.Compose([ transforms.Resize([224,224]), transforms.ToTensor(), transforms.Normalize((0.5, 0.5, 0.5), (0.5,.
原创
2021-03-02 14:35:05 ·
2006 阅读 ·
0 评论