Pytorch学习04_Transforms的使用

最新推荐文章于 2024-07-25 22:24:14 发布

写点什么呢

最新推荐文章于 2024-07-25 22:24:14 发布

阅读量855

点赞数 16

分类专栏：学习记录 Pytorch 文章标签： pytorch 学习人工智能 python pycharm

本文链接：https://blog.csdn.net/weixin_53384447/article/details/135782954

版权

学习记录同时被 2 个专栏收录

21 篇文章 4 订阅

订阅专栏

Pytorch

12 篇文章 0 订阅

订阅专栏

本文介绍了如何在PyCharm中利用Structure功能查看模块定义，以及如何使用PIL库将图像转换为PyTorch的Tensor类型，包括ToTensor的作用和TensorBoard的图像添加。最后提到结合其他模块实践观察Tensor效果，并给出了相关教程链接。

摘要由CSDN通过智能技术生成

Structure的使用

在Pycharm的左下角可以点开structure，查看一些模块内的函数定义，没有的话，去到设置里搜索“keymap”,搜索”sturcture“，可以找到对应的快捷键

查看图片

from PIL import Image
from torchvision import transforms
#python-tensor数据类型
# transforms.ToTensor解决两个问题
# 1、transforms能解决什么问题
# 2、为什么需要Tensor数据类型
# 绝对路径 W:\Projects\Pytorch01\torch_learn01\dataset\train\ants_image\0013035.jpg
# 相对路径 dataset\train\ants_image\0013035.jpg
img_path="dataset/train/ants_image/0013035.jpg"
img=Image.open(img_path)
print(img)

运行后可以看到输出结果

类型 RGB模式，尺寸是768x512，at逻辑地址

PIL 图像转换为 PyTorch 张量

from PIL import Image
from torchvision import transforms
#python-tensor数据类型
# transforms.ToTensor解决两个问题
# 1、transforms能解决什么问题
# 2、为什么需要Tensor数据类型
# 绝对路径 W:\Projects\Pytorch01\torch_learn01\dataset\train\ants_image\0013035.jpg
# 相对路径 dataset\train\ants_image\0013035.jpg
img_path="dataset/train/ants_image/0013035.jpg"
img=Image.open(img_path)
# print(img)
tensor_trans=transforms.ToTensor()# 创建 ToTensor 转换
tensor_img=tensor_trans(img)# 将 PIL 图像转换为 PyTorch 张量
print(tensor_img)

理解

from PIL import Image
from torch.utils.tensorboard import SummaryWriter
from torchvision import transforms
#python-tensor数据类型
# transforms.ToTensor解决两个问题
# 2、为什么需要Tensor数据类型
img_path="dataset/train/ants_image/0013035.jpg"
img=Image.open(img_path)

writer=SummaryWriter("logs")


tensor_trans=transforms.ToTensor()# 创建 ToTensor 转换
tensor_img=tensor_trans(img)# 将 PIL 图像转换为 PyTorch 张量
writer.add_image("Tensor_img",tensor_img)

writer.close()

终端运行