python自动计算图像数据集的RGB均值
更新时间:2021年06月18日 16:59:32 作者:南石北岸生
这篇文章主要为大家详细介绍了python自动计算图像数据集的RGB均值,文中示例代码介绍的非常详细,具有一定的参考价值,感兴趣的小伙伴们可以参考一下
本文实例为大家分享了python自动计算图像数据集的RGB均值,供大家参考,具体内容如下
图像数据集往往要进行去均值,以保证更快的收敛。
代码:
创建一个mean.py,写入如下代码。修改路径即可使用
'''
qhy
2018.12.3
'''
import os
import numpy as np
import cv2
ims_path='C:/Users/my/Desktop/JPEGImages/'# 图像数据集的路径
ims_list=os.listdir(ims_path)
R_means=[]
G_means=[]
B_means=[]
for im_list in ims_list:
im=cv2.imread(ims_path+im_list)
#extrect value of diffient channel
im_R=im[:,:,0]
im_G=im[:,:,1]
im_B=im[:,:,2]
#count mean for every channel
im_R_mean=np.mean(im_R)
im_G_mean=np.mean(im_G)
im_B_mean=np.mean(im_B)
#save single mean value to a set of means
R_means.append(im_R_mean)
G_means.append(im_G_mean)
B_means.append(im_B_mean)
print('图片:{} 的 RGB平均值为 \n[{},{},{}]'.format(im_list,im_R_mean,im_G_mean,im_B_mean) )
#three sets into a large set
a=[R_means,G_means,B_means]
mean=[0,0,0]
#count the sum of different channel means
mean[0]=np.mean(a[0])
mean[1]=np.mean(a[1])
mean[2]=np.mean(a[2])
print('数据集的BGR平均值为\n[{},{},{}]'.format( mean[0],mean[1],mean[2]) )
#cv.imread()读取Img时候将rgb转换为了bgr,谢谢taylover-pei的修正。
终端运行: python mean.py
结果示例如下:

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持脚本之家。
相关文章
Python爬虫框架scrapy实现downloader_middleware设置proxy代理功能示例
这篇文章主要介绍了Python爬虫框架scrapy实现downloader_middleware设置proxy代理功能,结合实例形式分析了scrapy框架proxy代理设置技巧与相关问题注意事项,需要的朋友可以参考下2018-08-08
Tensorflow中使用tfrecord方式读取数据的方法
这篇文章主要介绍了Tensorflow中使用tfrecord方式读取数据的方法,适用于数据较多时,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看看吧2018-06-06
Python中getattr函数和hasattr函数作用详解
这篇文章主要介绍了Python中getattr函数和hasattr函数作用的相关知识,非常不错具有参考借鉴价值,需要的朋友可以参考下2016-06-06


最新评论