torchvision的transforms的使用总结
import torch
from torchvision import transforms
from PIL import Image
image=Image.open('dog.jpg')#这张图片假如是rgb三通道的,分辨率是180*180
print(image.size)
toTensor=transforms.ToTensor()# 实例化一个toTensor
image_tensor=toTensor(image)
print(image_tensor.size())#这里就会输出[3,180,180]
#一般我们训练好的模型去detect需要把这个图像转成4维度
image_tensor=image_tensor.reshape(1,3,180,180)