pytorch VGG16预训练分类模型识别Cat
import torch
import torch.nn.functional as F
import torchvision.transforms
import torchvision.models as models
import PIL.Image as Image
import numpy as np
image = Image.open('D:/torch/VGG16/cat.jpg')
print(image.size, image.format, image.mode)
crop_obj.
原创
2020-12-11 09:49:46 ·
1974 阅读 ·
0 评论