albumentations 的数据增强为什么是 先 Normalize, 再 ToTensorV2,而 pytorch 正好相反

发布时间 2023-07-12 18:28:16作者: cold_moon

albumentations:

T += [A.Normalize(mean=mean, std=std), ToTensorV2()]  # Normalize and convert to Tensor

torchvision:

T.ToTensor(),
T.Normalize(IMAGENET_MEAN, IMAGENET_STD),

原因: A.Normalize 已经包含了 将 8 位图像 (0-255) 转换为 (0-1)(将 mean 和 std x 255,然后再做 减均值,除以 std), ToTensorV2() 只是简单的转为了 tensor.

而 torchvision 中,T.ToTensor() 会先转为 tensor,再将将 8 位图像 (0-255) 转换为 (0-1)