首页 > 其他分享 >albumentations 的数据增强为什么是 先 Normalize, 再 ToTensorV2,而 pytorch 正好相反

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

时间:2023-07-12 18:33:43浏览次数:38  
标签:Normalize std ToTensorV2 albumentations 255 mean

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)

标签:Normalize,std,ToTensorV2,albumentations,255,mean
From: https://www.cnblogs.com/odesey/p/17548515.html

相关文章