2000字范文,分享全网优秀范文,学习好帮手!
2000字范文 > torchvision.transforms.ToTensor的用法

torchvision.transforms.ToTensor的用法

时间:2023-01-11 04:38:49

相关推荐

torchvision.transforms.ToTensor的用法

#torchvision.transforms.ToTensor

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

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

img.tobytes() 将图片转化成内存中的存储格式

torch.BytesStorage.frombuffer(img.tobytes() ) 将字节以流的形式输入,转化成一维的张量

对张量进行reshape

对张量进行transpose

将当前张量的每个元素除以255

输出张量

##torchvision.transforms.ToPILImage#

对于一个Tensor的转化过程是:

将张量的每个元素乘上255

将张量的数据类型有FloatTensor转化成Uint8

将张量转化成numpy的ndarray类型

对ndarray对象做transpose (1, 2, 0)的操作

利用Image下的fromarray函数,将ndarray对象转化成PILImage形式

输出PILImage

————————————————

版权声明:本文为CSDN博主「啧啧啧biubiu」的原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接及本声明。

原文链接:/qq_37385726/article/details/81811466

本内容不代表本网观点和政治立场,如有侵犯你的权益请联系我们处理。
网友评论
网友评论仅供其表达个人看法,并不表明网站立场。