ToTensor的使用————Transforms

哔哩大学的PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】
的P12讲讲述了ToTensor的使用。
代码及运行结果为:

from PIL import Image
from torch.utils.tensorboard import SummaryWriter#从tensorboard引入SummaryWriter
from torchvision import transforms

#  之前说过tensorboard必须为tensor的数据类型
# tensor实际上就是一个多维数组,能够创造更高维度的矩阵、向量,
# 具体参考知乎   https://zhuanlan.zhihu.com/p/48982978
# class ToTensor用法:
#     """Convert a ``PIL Image`` or ``numpy.ndarray`` to tensor. This transform does not support torchscript.
# 输入必须为PIL Image,或者numpy.ndarray,转化为tensor类型。

writer = SummaryWriter("logs")#首先把tensorboard做一个简单的配置
img = Image.open("images/pytorch.png")
print(img)

#用transforms中的一个totensor数据类型,起名叫trans_totensor,创建这样一个对象
trans_totensor =  transforms.ToTensor()
img_tensor = trans_totensor(img)  #调用的一个方法,把上边的img变成了一个tensor类型
#,之后img可以放到tensorboard中
writer.add_image("ToTensor", img_tensor)#在括号里按住ctrl+P后,tag就叫ToTensor,img就是img_tensor
writer.close()#一个关闭

ToTensor的使用————Transforms_第1张图片
ToTensor的使用————Transforms_第2张图片

你可能感兴趣的:(pytorch,pytorch,深度学习,神经网络)