pytorch学习笔记（3）

最新推荐文章于 2024-01-31 22:02:27 发布

不瘦回150不改

最新推荐文章于 2024-01-31 22:02:27 发布

阅读量768

点赞数

文章标签： pytorch 深度学习 python

本文链接：https://blog.csdn.net/qq_61407913/article/details/122755320

版权

from tensorboardX import SummaryWriter
from torchvision import transforms
from PIL import Image#导入一些库

#python的用法-》 tensor数据类型
#通过 transforms.ToTensor去看两个问题
#1、 transforms该如何使用(python)
#2、为什么我们需要Tensor数据类型

img_path="data/train/ants_image/0013035.jpg"#复制相对路径
img = Image.open(img_path)#打开文件

writer = SummaryWriter("logs")#SummaryWriter

tensor_trans = transforms.ToTensor()#通过transforms里面的ToTensor（）来进行一个转化
tensor_img = tensor_trans(img)

writer.add_image("Tensor_img",tensor_img)#add一下图片
writer.close()#关闭一下writer

今天学习了transform中的一些转化图片的一些工具，主要就是ToTensor，转换成神经网络常用的Tensor形式，明天继续加油！