天天看點

圖像增強(直方圖均衡化、拉普拉斯、Log、Gamma)

#include <opencv2/imgproc/imgproc.hpp>    
#include <opencv2/highgui/highgui.hpp>
#include<opencv2/opencv.hpp>
#include <iostream>

using namespace cv;

//OpenCV圖像增強算法實作(直方圖均衡化、拉普拉斯、Log、Gamma)

void hisitogramEhance(Mat& image);//基于直方圖均衡化的圖像增強
void laplaceEhance(Mat& image);//基于拉普拉斯算子的圖像增強
void logEhance(Mat& image);//基于對數Log變換的圖像增強
void gamaEhance(Mat& image); //基于伽馬變換的圖像增強

int main(int argc, char *argv[])
{
	Mat image = imread("1.jpg", 1);
	if (image.empty())
	{
		std::cout << "打開圖檔失敗,請檢查" << std::endl;
		return -1;
	}
	imshow("原圖像", image);
	int key = 0;
	while (1)
	{
		key = waitKey(9);//讀取鍵值到key變量中
		switch (key)
		{
		case 27:
			return 0; 
			break;
		case 'q':
			hisitogramEhance(image); 
			break;
		case 'w':
			laplaceEhance(image); 
			break;
		case 'e':
			logEhance(image);
			break;
		case 'r':
			gamaEhance(image); 
			break;
		
		}
	}
	
	waitKey();
	return 0;
}

void hisitogramEhance(Mat& image)
{
	Mat imageRGB[3];
	split(image, imageRGB);
	for (int i = 0; i < 3; i++)
	{
		equalizeHist(imageRGB[i], imageRGB[i]);
	}
	merge(imageRGB, 3, image);
	imshow("直方圖均衡化圖像增強效果", image);
}
void laplaceEhance(Mat& image)
{
	Mat imageEnhance;
	Mat kernel = (Mat_<float>(3, 3) << 0, -1, 0, 0, 5, 0, 0, -1, 0);
	filter2D(image, imageEnhance, CV_8UC3, kernel);
	imshow("拉普拉斯算子圖像增強效果", imageEnhance);
}
void logEhance(Mat& image)
{
	Mat imageLog(image.size(), CV_32FC3);

	for (int i = 0; i < image.rows; i++)
	{
		for (int j = 0; j < image.cols; j++)
		{
			imageLog.at<Vec3f>(i, j)[0] = log(1 + image.at<Vec3b>(i, j)[0]);
			imageLog.at<Vec3f>(i, j)[1] = log(1 + image.at<Vec3b>(i, j)[1]);
			imageLog.at<Vec3f>(i, j)[2] = log(1 + image.at<Vec3b>(i, j)[2]);
		}
	}
	//歸一化到0~255  
	normalize(imageLog, imageLog, 0, 255, CV_MINMAX);
	//轉換成8bit圖像顯示  
	convertScaleAbs(imageLog, imageLog);
	//imshow("Soure", image);
	imshow("Log變換圖像增強效果", imageLog);

}

void gamaEhance(Mat& image)
{
	Mat imageGamma(image.size(), CV_32FC3);
	for (int i = 0; i < image.rows; i++)
	{
		for (int j = 0; j < image.cols; j++)
		{
			imageGamma.at<Vec3f>(i, j)[0] = (image.at<Vec3b>(i, j)[0])*(image.at<Vec3b>(i, j)[0])*(image.at<Vec3b>(i, j)[0]);
			imageGamma.at<Vec3f>(i, j)[1] = (image.at<Vec3b>(i, j)[1])*(image.at<Vec3b>(i, j)[1])*(image.at<Vec3b>(i, j)[1]);
			imageGamma.at<Vec3f>(i, j)[2] = (image.at<Vec3b>(i, j)[2])*(image.at<Vec3b>(i, j)[2])*(image.at<Vec3b>(i, j)[2]);
		}
	}
	//歸一化到0~255  
	normalize(imageGamma, imageGamma, 0, 255, CV_MINMAX);
	//轉換成8bit圖像顯示  
	convertScaleAbs(imageGamma, imageGamma);
	imshow("伽馬變換圖像增強效果", imageGamma);

}