2021-03-22

最新推荐文章于 2024-09-22 17:12:26 发布

小机灵鬼~

最新推荐文章于 2024-09-22 17:12:26 发布

阅读量162

点赞数

文章标签： python

本文链接：https://blog.csdn.net/qq_37279375/article/details/115066908

版权

该博客介绍了使用PyTorch进行RGB图像预处理的方法，包括随机水平翻转、调整尺寸、随机裁剪和归一化。预处理的均值和标准差取值来源于ImageNet数据集，这些值用于标准化图像，提高模型训练效果。了解如何为特定数据集计算这些数值，以优化模型性能。

摘要由CSDN通过智能技术生成

RGB图像训练前的预处理（pytorch）：

train_transforms = torchvision.transforms.Compose([
            torchvision.transforms.RandomHorizontalFlip(),
            torchvision.transforms.Resize((384, 288)),
            torchvision.transforms.RandomCrop((288, 216)),
            torchvision.transforms.ToTensor(),
            torchvision.transforms.Normalize(mean=(0.485, 0.456, 0.406),
                                             std=(0.229, 0.224, 0.225))])

见到的对RGB图像进行均值标准差预处理的设定值是怎么来的呢？

ImageNet数据集中计算得到的

专门针对一个数据库里面的图像如何计算均值和标准差呢？