Transform的作用
把图片经过Transforms的一些函数之后就会对图片进行一些变化。比如,resize就是改变其大小,totensor就是把图片PIL或者numpy类型转化为Tensor类型。
Transforms的结构及和 用法
totensor的使用
Transforms下的toTensor是一个对象,我们需要先实例化一个对象之后再通过这个实例化的对象进行调用。
from torchvision import transforms
tensor_class = transforms.ToTensor()
img_tensor = tensor_class(img_PIL)
为什么使用transforms转化为Tensor类型
因为tensor类型包装了我们神经网络中所需要的一些参数,而PIL或者numpy类型是没有的。
ToTensor实例代码
from PIL import Image
from torchvision.transforms import transforms
from torch.utils.tensorboard import SummaryWriter
img_PIL = Image.open("Dataset/hymenoptera_data/train/ants/0013035.jpg")
# print(img_PIL)
tensor_class = transforms.ToTensor()
img_tensor = tensor_class(img_PIL)
writer = SummaryWriter('logs')
writer.add_image('tensor-test',img_tensor,1,)
# print(img_tensor)
writer.close()
效果如下