transforms.Normalize，计算数据量大数据集的像素均值（mean)和标准差（std)

最新推荐文章于 2023-08-09 18:39:22 发布

划水的鱼儿

最新推荐文章于 2023-08-09 18:39:22 发布

阅读量1.8k

点赞数 6

分类专栏：深度学习工具文章标签：深度学习 pytorch

本文链接：https://blog.csdn.net/Cai_Xu_Kun/article/details/108718584

版权

代码来源: https://blog.csdn.net/AlfredXXX/article/details/102852560
对大量数据在训练前做出标准化处理
均值标准差定义：
均值反映了图像的亮度，均值越大说明图像亮度越大，反之越小；
标准差反映了图像像素值与均值的离散程度，标准差越大说明图像的质量越好；

import os
import cv2
import numpy as np

path = '/home/train/images'       #图片保存路径

def compute(path):
    file_names = os.listdir(path)
    per_image_Rmean = []
    per_image_Gmean = []
    per_image_Bmean = []
    for file_name