“毛星云OpenCV3编程入门之python实现”第九篇亮度、对比度

5.4亮度、对比度

python代码：

# -*- coding: utf-8 -*-

import cv2
import numpy as np

gcontrastvalue = 80      # 对比度
gbrightvalue = 80        # 亮度


def contrast(contrastvalue):
	desImage = srcImage.copy()
	global gcontrastvalue
	global gbrightvalue
	print('gcontrastvalue：', gcontrastvalue)
	print('gbrightvalue：', gbrightvalue)
	table = []
	for i in range(256):
		data = int(i * gcontrastvalue * 0.01 + gbrightvalue)
		if data < 0:
			table.append(0)
		elif data > 255:
			table.append(255)
		else:
			table.append(data)
	table = np.array(table, dtype=np.uint8)  # 将数组转换为映射矩阵，数据类型与原图像数据保持一致
	cv2.LUT(srcImage, table, desImage)  # 利用系统函数LUT进行映射矩阵替换原图形数据
	cv2.imshow("window1", desImage)
	gcontrastvalue = contrastvalue


def bright(brightvalue):
	desImage = srcImage.copy()
	global gcontrastvalue
	global gbrightvalue
	print('gcontrastvalue：', gcontrastvalue)
	print('gbrightvalue：', gbrightvalue)
	table = []
	for i in range(256):
		data = int(i * gcontrastvalue * 0.01 + gbrightvalue)
		if data < 0:
			table.append(0)
		elif data > 255:
			table.append(255)
		else:
			table.append(data)

	table = np.array(table, dtype=np.uint8)  # 将数组转换为映射矩阵，数据类型与原图像数据保持一致
	cv2.LUT(srcImage, table, desImage)  # 利用系统函数LUT进行映射矩阵替换原图形数据
	cv2.imshow("window1", desImage)
	gbrightvalue = brightvalue


srcImage = cv2.imread("E:/Study/python/OpenCV_study/img/8.jpg")


# print(srcImage.shape)             # 获取原图像的尺寸
# print(type(srcImage[0][0][0]))    # 获取原图像的数据类型，为后面做映射矩阵服务

cv2.namedWindow("window1")
cv2.createTrackbar("对比度", "window1", 80, 300, contrast)
cv2.createTrackbar("亮  度", "window1", 80, 100, bright)
cv2.imshow("window1", srcImage)
cv2.waitKey(0)

原书中的C++代码：

//--------------------------------------【程序说明】-------------------------------------------
//		程序说明：《OpenCV3编程入门》OpenCV3版书本配套示例程序27
//		程序描述：图像对比度、亮度值调整
//		开发测试所用IDE版本：Visual Studio 2010
//		开发测试所用OpenCV版本：	3.0 beta
//		2014年11月 Created by @浅墨_毛星云
//		2014年12月 Revised by @浅墨_毛星云
//------------------------------------------------------------------------------------------------



//-----------------------------------【头文件包含部分】---------------------------------------
//	描述：包含程序所依赖的头文件
//---------------------------------------------------------------------------------------------- 
#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include "opencv2/imgproc/imgproc.hpp"
#include <iostream>

//-----------------------------------【命名空间声明部分】---------------------------------------
//	描述：包含程序所使用的命名空间
//-----------------------------------------------------------------------------------------------   
using namespace std;
using namespace cv;


//-----------------------------------【全局函数声明部分】--------------------------------------
//	描述：全局函数声明
//-----------------------------------------------------------------------------------------------
static void ContrastAndBright(int, void *);
void   ShowHelpText();

//-----------------------------------【全局变量声明部分】--------------------------------------
//	描述：全局变量声明
//-----------------------------------------------------------------------------------------------
int g_nContrastValue; //对比度值
int g_nBrightValue;  //亮度值
Mat g_srcImage,g_dstImage;
//-----------------------------------【main( )函数】--------------------------------------------
//	描述：控制台应用程序的入口函数，我们的程序从这里开始
//-----------------------------------------------------------------------------------------------
int main(   )
{
	//改变控制台前景色和背景色
	system("color 2F");  

	ShowHelpText();
	// 读入用户提供的图像
	g_srcImage = imread( "1.jpg");
	if( !g_srcImage.data ) { printf("读取g_srcImage图片错误~！ \n"); return false; }
	g_dstImage = Mat::zeros( g_srcImage.size(), g_srcImage.type() );

	//设定对比度和亮度的初值
	g_nContrastValue=80;
	g_nBrightValue=80;

	//创建窗口
	namedWindow("【效果图窗口】", 1);

	//创建轨迹条
	createTrackbar("对比度：", "【效果图窗口】",&g_nContrastValue, 300,ContrastAndBright );
	createTrackbar("亮   度：", "【效果图窗口】",&g_nBrightValue, 200,ContrastAndBright );

	//调用回调函数
	ContrastAndBright(g_nContrastValue,0);
	ContrastAndBright(g_nBrightValue,0);

	//输出一些帮助信息
	cout<<endl<<"\t运行成功，请调整滚动条观察图像效果\n\n"
		<<"\t按下“q”键时，程序退出\n";

	//按下“q”键时，程序退出
	while(char(waitKey(1)) != 'q') {}
	return 0;
}




//-----------------------------------【ShowHelpText( )函数】----------------------------------
//		 描述：输出一些帮助信息
//----------------------------------------------------------------------------------------------
void ShowHelpText()
{
	//输出欢迎信息和OpenCV版本
	printf("\n\n\t\t\t非常感谢购买《OpenCV3编程入门》一书！\n");
	printf("\n\n\t\t\t此为本书OpenCV3版的第27个配套示例程序\n");
	printf("\n\n\t\t\t   当前使用的OpenCV版本为：" CV_VERSION );
	printf("\n\n  ----------------------------------------------------------------------------\n");
}


//-----------------------------【ContrastAndBright( )函数】------------------------------------
//	描述：改变图像对比度和亮度值的回调函数
//-----------------------------------------------------------------------------------------------
static void ContrastAndBright(int, void *)
{

	// 创建窗口
	namedWindow("【原始图窗口】", 1);

	// 三个for循环，执行运算 g_dstImage(i,j) = a*g_srcImage(i,j) + b
	for( int y = 0; y < g_srcImage.rows; y++ )
	{
		for( int x = 0; x < g_srcImage.cols; x++ )
		{
			for( int c = 0; c < 3; c++ )
			{
				g_dstImage.at<Vec3b>(y,x)[c] = saturate_cast<uchar>( (g_nContrastValue*0.01)*( g_srcImage.at<Vec3b>(y,x)[c] ) + g_nBrightValue );
			}
		}
	}

	// 显示图像
	imshow("【原始图窗口】", g_srcImage);
	imshow("【效果图窗口】", g_dstImage);
}