PyTorch_4_Transforms的使用

from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms

# python 的用法->Tensor数据类型
# 通过transforms.ToTensor去解决两个问题

# 2、为什么我们需要Tensor的数据类型

# 绝对路径“H:\Python_Project\Pycharm_Anaconda\LearnPytorch\practice\train\ants_image\0013035.jpg”
# 相对路径“practice/train/ants_image/0013035.jpg”
img_path = "practice/train/ants_image/0013035.jpg"
img_path_abs = r"H:\Python_Project\Pycharm_Anaconda\LearnPytorch\practice\train\ants_image\0013035.jpg"
img = Image.open(img_path)
print(type(img))

writer = SummaryWriter("logs")

# 1、transforms该如何使用(python)
tensor_trains = transforms.ToTensor()
# 要使用的时候，直接把这个对象这样就行了：def __call__(self, pic):
tensor_img = tensor_trains(img)

writer.add_image("Tensor_img", tensor_img)
writer.close()

# print(type(tensor_img))
# print(tensor_img)