transform的结构及用法

---

一、transform作用

结构图1

结构图2

二、实际应用tensor类型

详细代码如下：

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

# python 的用法 -》 tensor数据类型
# 通过transform.ToTensor去看两个问题
# 2、为什么我们需要Tensor数据类型 :因为tensor类型数据包含了训练神经网络所需要的属性

image_path = r"D:\pycharm_professional\PycharmProjects\learning_pytorch\dataset\train\ants_image\0013035.jpg"

img = Image.open(image_path)

writer = SummaryWriter("logs") # ctrl+P查看参数   ctrl + alt解决爆红问题

# 1、transforms该如何使用（python）
# 生成一个对象
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)

# 使用tensorboard展示
writer.add_image("tensor_img",tensor_img,1)
writer.close()

注意： transforms是在哪导入的from torchvision import transforms

terminal终端输入指令tensorboard --logdir="logs" --port=6007

结果如下：

pycharm使用小技巧：
ctrl + P 可以显示参数，使用函数或类时，将光标定位到所使用方法的括号内，按住ctrl+p可以显示参数

alt + enter可以提供爆红的解决方案，将光标定位到划红线的位置，按住alt+enter即可提供爆红的解决方案。