python3+opencv 使用灰度直方圖來(lái)判斷圖片的亮暗操作

更新時(shí)間：2020年06月02日 11:31:12 作者：ZJE_ANDY

這篇文章主要介紹了python3+opencv 使用灰度直方圖來(lái)判斷圖片的亮暗操作，具有很好的參考價(jià)值，希望對(duì)大家有所幫助。一起跟隨小編過(guò)來(lái)看看吧

1.如何讓計(jì)算機(jī)自動(dòng)判斷一張圖是否偏暗?或是判斷一張圖是否是處于夜晚?我們可以先把圖片轉(zhuǎn)換為灰度圖,然后根據(jù)灰度值的分布來(lái)判斷,如:

我們可以從上圖看到,晚上的圖片的灰度值是集中在前段的,如0~30多左右,我們?cè)倏匆粡埍容^明亮的圖片:

明亮的圖片的灰度直方圖是比較靠后的.

因此要判斷圖片的亮暗,只需要統(tǒng)計(jì)偏暗的像素個(gè)數(shù),再除以圖片像素的總個(gè)數(shù),得到百分比p即可,至于p大于多少即判斷為暗,則可以由你自己設(shè)置.下面給出代碼,此代碼可以在裝滿圖片的目錄里選出偏暗的圖片,然后把這些圖片保存下來(lái).

2.代碼：

import cv2;
import numpy as np;
import matplotlib.pyplot as plt;
import os;
import sys;
 
def func(img,pic_path,pic):
	#把圖片轉(zhuǎn)換為灰度圖
	gray_img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY);
	#獲取灰度圖矩陣的行數(shù)和列數(shù)
	r,c = gray_img.shape[:2];
	dark_sum=0;	#偏暗的像素 初始化為0個(gè)
	dark_prop=0;	#偏暗像素所占比例初始化為0
	piexs_sum=r*c;	#整個(gè)弧度圖的像素個(gè)數(shù)為r*c
	
	#遍歷灰度圖的所有像素
	for row in gray_img:
		for colum in row:
			if colum<40:	#人為設(shè)置的超參數(shù),表示0~39的灰度值為暗
				dark_sum+=1;
	dark_prop=dark_sum/(piexs_sum);	
	print("dark_sum:"+str(dark_sum));
	print("piexs_sum:"+str(piexs_sum));
	print("dark_prop=dark_sum/piexs_sum:"+str(dark_prop));
	if dark_prop >=0.75:	#人為設(shè)置的超參數(shù):表示若偏暗像素所占比例超過(guò)0.78,則這張圖被認(rèn)為整體環(huán)境黑暗的圖片
		print(pic_path+" is dark!");
		cv2.imwrite("../DarkPicDir/"+pic,img);#把被認(rèn)為黑暗的圖片保存
	else:
		print(pic_path+" is bright!")
	#hist(pic_path); #若要查看圖片的灰度值分布情況,可以這個(gè)注釋解除
 
#用于顯示圖片的灰度直方圖
def hist(pic_path):
	img=cv2.imread(pic_path,0);
	hist = cv2.calcHist([img],[0],None,[256],[0,256])
	plt.subplot(121)
	plt.imshow(img,'gray')
	plt.xticks([])
	plt.yticks([])
	plt.title("Original")
	plt.subplot(122)
	plt.hist(img.ravel(),256,[0,256])
	plt.show()
	
#讀取給定目錄的所有圖片
def readAllPictures(pics_path):
	if not os.path.exists(pics_path):
		print("路徑錯(cuò)誤，路徑不存在！")
		return;
	allPics = [];
	pics = os.listdir(pics_path);
	for pic in pics:
		pic_path = os.path.join(pics_path,pic);
		if os.path.isfile(pic_path):
			allPics.append(pic_path);
			img=cv2.imread(pic_path);
			func(img,pic_path,pic);
	return allPics;
 
#創(chuàng)建用于存放黑暗圖片的目錄
def createDarkDir():
	DarkDirPath = "../DarkPicDir";
	isExists = os.path.exists(DarkDirPath);
	if not isExists:
		os.makedirs(DarkDirPath);
		print("dark pics dir is created successfully!");
		return True;
	else:
		return False;
 
if __name__ =='__main__':
	pics_path = sys.argv[1];#獲取所給圖片目錄
	createDarkDir();
	allPics=readAllPictures(pics_path);

運(yùn)行命令: python3 cal.py 圖片集的路徑名

(其中cal.py 是python代碼的文件名)

運(yùn)行結(jié)果: