问题遇到的现象和发生背景
关于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,进行归一化处理