深度学习之用python计算数据集均值

import os  

from PIL import Image    

import matplotlib.pyplot as plt  

import numpy as np  

from scipy.misc import imread   


filepath ='/home/JPEGImages' # 数据集目录  

pathDir = os.listdir(filepath)  


R_channel =0  

G_channel =0  

B_channel =0  

for idx in xrange(len(pathDir)):  

    filename = pathDir[idx]  

    img = imread(os.path.join(filepath, filename))  

R_channel = R_channel + np.sum(img[:,:,0])  

G_channel = G_channel + np.sum(img[:,:,1])  

B_channel = B_channel + np.sum(img[:,:,2])  


num = len(pathDir) *384 * 512 # 这里(384,512)是每幅图片的大小,所有图片尺寸都一样  

R_mean = R_channel / num  

G_mean = G_channel / num  

B_mean = B_channel / num  


print("R_mean is %f, G_mean is %f, B_mean is %f" %(R_mean, G_mean, B_mean))  

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容