pytorch1.1.0-python3.6-CUDA9.0-ToTensor ToPILImage

匿名 (未验证) 提交于 2019-12-02 22:51:30

参考:

PIL.Image和np.ndarray图片与Tensor之间的转换

https://blog.csdn.net/tsq292978891/article/details/78767326

Pytorch之深入torchvision.transforms.ToTensor与ToPILImage

https://blog.csdn.net/qq_37385726/article/details/81811466

PyTorch载入图片后ToTensor解读(含PIL和OpenCV读取图片对比)

https://www.cnblogs.com/ocean1100/p/9494640.html

torchvision.transforms.ToTensor
对于一个图片img,调用ToTensor转化成张量的形式,发生的不是将图片的RGB三维信道矩阵变成tensor

图片在内存中以bytes的形式存储,转化过程的步骤是:

  • 对张量进行reshape
  • 对张量进行transpose
  • 将当前张量的每个元素除以255
  • 输出张量

torchvision.transforms.ToPILImage
对于一个Tensor的转化过程是:

  • 将张量的每个元素乘上255
  • 将张量的数据类型有FloatTensor转化成Uint8
  • 将张量转化成numpy的ndarray类型
  • 利用Image下的fromarray函数,将ndarray对象转化成PILImage形式
  • 输出PILImage

文章来源: https://blog.csdn.net/baidu_40840693/article/details/92011333
标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!