其他分享
首页 > 其他分享> > 计算pytorch标准化(Normalize)所需要数据集的均值和方差

计算pytorch标准化(Normalize)所需要数据集的均值和方差

作者:互联网

先说明一下情况

1、如果是自己的数据集,mean 和 std 肯定要在normalize之前自己先算好再传进去的

2、有两种情况:
a)数据集在加载的时候就已经转换成了[0, 1].
b)应用了torchvision.transforms.ToTensor,其作用是
( Converts a PIL Image or numpy.ndarray (H x W x C) in the range [0, 255] to a torch.FloatTensor of shape (C x H x W) in the range [0.0, 1.0] )

3、[0.485, 0.456, 0.406]这一组平均值一般是抽样算出来的。
————————————————
版权声明:本文为CSDN博主「茗君(Major_s)」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/qq_41375318/article/details/105912138

标签:Normalize,方差,声明,range,pytorch,原文,版权,数据,链接
来源: https://blog.csdn.net/weixin_38530843/article/details/121223343