使用pytorch的dataload方式计算自己的图片数据集的均值和标准差

网上看到一个使用opencv读取图片然后计算数据集的均值和标准差的,但是那个读取图片后把图片的每个值append到一个列表,要是数据集大的话内存真的会爆掉的啊,所以借助网上另一个使用pytorch的数据读取方式来计算的,原文https://www.cnblogs.com/wanghui-garcia/p/11448460.html 这篇是分别计算了训练集、测试集和验证集数据的均值和标准差并将均值和标准差保存到了一个文件中,我不需要那样子,我只需要计算我总数据集的均值标准差并输出就好了,所以做了一点修改。 首先说一下我的文件夹格式,没有分训练集测试集啥的,就是一个文件夹下面分类别放 ‘/home