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]
Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐