pytorch -- ToTensor使用

斑驳的岁月

于 2024-02-25 20:14:22 发布

阅读量1k

点赞数 6

分类专栏： pytorch 文章标签： pytorch 人工智能 python

本文链接：https://blog.youkuaiyun.com/weixin_39467231/article/details/136286669

版权

pytorch 专栏收录该内容

11 篇文章

订阅专栏

本文介绍了如何使用PyTorch中的`ToTensor`函数将PIL图像或numpy数组转换为张量，并展示了如何结合TensorBoard进行图像可视化。作者通过实例展示了如何加载图像、应用转换并将其添加到TensorBoard日志中。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

1. ToTensor定义
导入：from torchvision import transforms
通过transforms.ToTensor解决两个问题（PIL image/numpy.ndarray 转化为 tensor ）
在这里插入图片描述
ToTensor()返回一个ToTensor的对象(创建具体的工具)，传入pic就会返回一个Tensor类型的图片（使用工具）

ToTensor使用（结合tensorboard）

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

img_path =  'dataset/train/ants/6743948_2b8c096dda.jpg'
img = Image.open(img_path)
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img)
print(tensor_img)

writer = SummaryWriter("logs")
writer.add_image("Tensor_img",tensor_img)
writer.close()