哔哩大学的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()#一个关闭