PyTorch快速入门教程【小土堆】之Transforms的使用
视频地址P9. Transforms的使用(一)_哔哩哔哩_bilibili
本文中的transforms指的并不是那个知名的模型,而是Transforms.py所在的一个包,相当于一个工具箱,可以被调用,里面有Resize,PILToTensor等方法,可直接对图片进行操作
from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
# python的用法 -> tensor数据类型
# 通过transforms.ToTensor去看两个问题
# 绝对路径 C:\Users\chen\Desktop\PythonProject\data\train\ants_image\0013035.jpg
# 相对路径 data/train/ants_image/0013035.jpg
image_path = "data/train/ants_image/0013035.jpg"
img = Image.open(image_path)
writer = SummaryWriter("logs")
# 1、 transforms该如何使用(python)
tensor_trans = transforms.ToTensor() # 从transforms选择一个class进行创建
tensor_img = tensor_trans(img) #将img转换为了tensor类型的img
writer.add_image("Tensor_img", tensor_img)
writer.close()
# 2、为什么我们需要Tensor数据类型
# tensor类型中包含梯度,反向传播等深度学习需要常见的信息