pytorch快速入门（一）torchvision中的transforms（1）

解决两个问题

1）transforms()如何使用

2）为什么需要tensor数据类型

1、transforms结构及用法[以ToTensor()为例]

问题1）transforms()如何使用

 transforms.ToTensor()；作用  将“PIL Image”，“numpy.ndarray”转换为tensor

2、代码实例 

1、在pycharm中输入下列代码

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

img_path = "E:/PycharmProjects/Pytoch_learning/dataset/data/train/ants_image/5650366_e22b7e1065.jpg"
img = Image.open(img_path) #打开图片
print(img)
#>>输出结果：

writer = SummaryWriter(log_dir="E:/PycharmProjects/runs/flower_experiment") #生成一个写日志的writer，并将当前的计算图写入日志，生成的文件在当前文件夹下



# #问题1：transforms（）的使用
tensor_trans = transforms.ToTensor()
tensor_img = tensor_trans(img) #将PIL格式的img，转换成tensor
print(type(tensor_img))
# #>>输出结果：

writer.add_image("Tensor_img", tensor_img)

writer.close() #关闭

部分代码说明

• img_path地址为下图圈出的路径（数据集在上篇笔记评论中Tensorboard(二)：add_image()_rensweet的博客-CSDN博客）

• SummaryWriter（）生成事件文件

2、单机右键-->run

出现下图结果

3、tensorboard可视化

1、 在terminal中输入下列代码：

tensorboard --logdir="E:\PycharmProjects\runs\flower_experiment"

2、回车，得到下图结果

3、点击蓝色链接，即可出现下图界面

问题2）为什么要使用tensor

笔记 | 什么是张量（tensor）& 深度学习 - 知乎 (zhihu.com)

这篇文章中两个视频讲的很好