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

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)

posted @   Zenith_Hugh  阅读(666)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 阿里最新开源QwQ-32B,效果媲美deepseek-r1满血版,部署成本又又又降低了!
· 单线程的Redis速度为什么快?
· SQL Server 2025 AI相关能力初探
· AI编程工具终极对决:字节Trae VS Cursor,谁才是开发者新宠?
· 展开说说关于C#中ORM框架的用法!

喜欢请打赏

扫描二维码打赏

微信打赏

点击右上角即可分享
微信分享提示