在使用Tensor`规范化处理的时候报下面这个错
from torchvision import transforms
import numpy as np
import torchvision
import torch
data = np.random.randint(0, 255, size=12)
img = data.reshape(2,2,3)
print(img)
print("*"*100)
transform1 = transforms.Compose([
transforms.ToTensor(), # range [0, 255] -> [0.0,1.0]
transforms.Normalize(mean = (10,10,10), std = (1,1,1)),
]
)
# img = img.astype('float')
norm_img = transform1(img) #进行规范化处理
print(norm_img)
添加上这句就可以,其实是给设置元素类型,看上面的提示也是这个意思