transform的结构及用法


一、transform作用

结构图1

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即可提供爆红的解决方案。

在这里插入图片描述

猜你喜欢

转载自blog.csdn.net/booze_/article/details/125417499