albumentations ToTensorV2 from albumentations.pytorch import ToTensorV2 ToTensorV2(p=1) # 只会[h, w, c] -> [c, h, w],不会将数据归一化到[0, 1] # 注意0是均值, 1是方差, 而归一化后的某个元素值仍可能大于1, 因为均值、方差归一化到[0, 1], 最大值不一定一会在[0, 1]