【无标题】

苏梓鑫鑫鑫

于 2024-07-25 15:46:14 发布

阅读量81

点赞数 3

文章标签：深度学习 pytorch 人工智能

本文链接：https://blog.csdn.net/auebud/article/details/140691993

版权

用python读取图片
图片相对路径和绝对路径已给出
运行结果是显示出来图片




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

# python的用法
# 通过 transforms.totensor去看两个问题
# 1.transforms应该如何使用（python）
# 2. 为什么tensor数据类型

#绝对路径 E:\pyxm\pythonProject1\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)

Writer = SummaryWriter("logs")

tensor_trans = transforms. ToTensor()
tensor_img = tensor_trans(img)



print(tensor_img)
writer.add_image("Tensor Image", tensor_img)
writer.close()