9Transforms的使用（二）

橙黄橘绿时_Eden

已于 2022-05-21 11:52:29 修改

阅读量241

点赞数 1

分类专栏：土堆-pytorch 文章标签： pytorch python 深度学习

于 2022-05-21 11:50:02 首次发布

本文链接：https://blog.csdn.net/qq_45583898/article/details/124896256

版权

土堆-pytorch 专栏收录该内容

30 篇文章 5 订阅

订阅专栏

该篇博客探讨了为何在神经网络中需要使用Tensor数据类型。通过使用`torch.utils.tensorboard.SummaryWriter`和`torchvision.transforms.ToTensor()`，作者展示了如何将图像转换为Tensor并将其添加到TensorBoard进行可视化。这揭示了Tensor在处理和展示数据时的重要性，特别是在深度学习框架中。

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

2、为什么需要Tensor数据类型
因为其包含了神经网络中所需要的一些参数，如
在这里插入图片描述
使用opencv打开的格式是numpy，也是transforms.ToTensor支持的格式

from torch.utils.tensorboard import SummaryWriter进行加载显示

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

# python中的用法 =》 tensor数据类型
# 通过  transforms.ToTensor去看两个问题

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

# 绝对路径 C:\Users\Eden\Desktop\tudui_pytorch\hymenoptera_data\train\ants_image\0013035.jpg
# 相对路径 hymenoptera_data/train/ants_image/0013035.jpg
img_path = "hymenoptera_data/train/ants_image/0013035.jpg"
img = Image.open(img_path)

writer = SummaryWriter("logs")


# 1、transforms该如何使用（python）
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)
print(tensor_img)

writer.add_image("Tensor_img",tensor_img)

writer.close()