这是python numpy的归一化:
mean = np.array([0.485, 0.456, 0.406], dtype=np.float32)
std = np.array([0.229, 0.224, 0.225], dtype=np.float32)
img = img.astype(np.float32)
img = (img/255.0 - mean) / std
img = img.transpose([2, 0, 1])
img = torch.Tensor(np.asarray([img]))
pytorch的归一化参考: