木叶清风666 2022-04-06 14:10 采纳率: 0%
浏览 391

pytorch深度学习中torchvision.transforms.ToTensor()使用问题

问题遇到的现象和发生背景

关于torchvision.transforms.ToTensor(),官方解释是把一个取值范围是[0,255]的PIL.Image或者shape为(H,W,C)的numpy.ndarray,转换成形状为[C,H,W],取值范围是[0,1.0]的torch.FloadTensor,但是在使用过程中却发现只是将数据转化成了tensor,并没有映射到0~1

问题相关代码,请勿粘贴截图

from torchvision import transforms
import numpy as np
data = np.random.randint(0, 255, size=300)
img = data.reshape(10,10,3)
print(img.shape)
img_tensor = transforms.ToTensor()(img) # 转换成tensor
print(img_tensor)
输出见下图所示

运行结果及报错内容

结果并未将数据映射到0~1,进行归一化处理

img

  • 写回答

2条回答 默认 最新

  • Love And Program 新星创作者: 人工智能技术领域 2022-04-06 14:39
    关注

    只有Ndarray是np.uint8 类型的时候,才会归一化

    img


    有用记得采纳哦

    评论 编辑记录

报告相同问题?

问题事件

  • 创建了问题 4月6日