学习pytorch4 transforms的使用

B站小土堆视频学习

常用类ToTensor

1. ToTensor如何使用

类实例化–>__call__方法调用
输出结果：

2. 为什么我们需要tensor数据类型

tensor数据类型包含了神经网络需要用到的一些参数

PIL数据类型

tensor数据类型

代码

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

# python的用法--》 tensor数据类型
# 通过 transforms.ToTensor去看两个问题
# 1. transforms该如何使用（python）

img_path = "hymenoptera_data/train/bees/16838648_415acd9e3f.jpg"
img = Image.open(img_path)
img_trans = transforms.ToTensor()
img_tensor = img_trans(img)

print(img)
print(img_tensor)

writer = SummaryWriter("logs")
writer.add_image("tensor data", img_tensor)
writer.close()

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

tensorboard 启动命令

tensorboard --logdir=logs