利用OpenCV的函数inpaint()实现图像的污点修复

1
2
3
4
5
6
void cv::inpaint(	InputArray 	src,
					InputArray 	inpaintMask,
					OutputArray 	dst,
					double 	inpaintRadius,
					int 	flags
				)

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
//博主微信/QQ 2487872782
//有问题可以联系博主交流
//有图像处理需求也可联系博主
//图像处理技术交流QQ群 271891601

//OpenCV版本：3.0
//VS版本：2012

#include <opencv2/core/core.hpp>
#include <opencv2/highgui/highgui.hpp>
#include<opencv2/imgcodecs/imgcodecs.hpp>
#include <opencv2/imgproc/imgproc.hpp>
#include <opencv2/photo/photo.hpp>

#include <iostream>

using namespace cv;
using namespace std;


bool g_bDrawing = false;
Point g_CurrPoint, g_OrgPoint;
int g_nThick = 3, g_nBlue = 255, g_nGreen = 255, g_nRed = 0;
Mat srcImage;
Mat maskImage;

void onMouse(int event, int x, int y, int flag, void *param)
{
	switch (event)
	{
		//移动鼠标的时候  
	case CV_EVENT_MOUSEMOVE:
	{
		g_OrgPoint = g_CurrPoint;
		g_CurrPoint = Point(x, y);

		if (g_bDrawing == 1)
		{
			line(srcImage, g_CurrPoint, g_OrgPoint, Scalar(g_nBlue, g_nGreen, g_nRed), g_nThick);
			imshow("【原图-可用鼠标绘制污点】", srcImage);

			//在掩膜图上进行显示  
			line(maskImage, g_CurrPoint, g_OrgPoint, Scalar(g_nBlue, g_nGreen, g_nRed), g_nThick);
			imshow("【掩膜图像】", maskImage);
		}
	}
		break;
		//点击鼠标左键时  
	case CV_EVENT_LBUTTONDOWN:
	{
		g_bDrawing = true;
		g_OrgPoint = Point(x, y);
		g_CurrPoint = g_OrgPoint;
	}
		break;
		//松开鼠标左键时  
	case CV_EVENT_LBUTTONUP:
	{
		g_bDrawing = false;
	}
		break;
	}
}

int main()
{

	srcImage = imread("F:/material/images/P0005-BaoXiaofeng-02.jpg");

	//用一个Mat对象来存储原图像  
	Mat g_srcImage;
	srcImage.copyTo(g_srcImage);

	//为掩膜图分配空间  
	maskImage.create(srcImage.size(), CV_8UC1);
	maskImage = Scalar::all(0);

	namedWindow("【原图-可用鼠标绘制污点】");
	setMouseCallback("【原图-可用鼠标绘制污点】", onMouse, 0);


	char key;
	while (1)
	{
		imshow("【原图-可用鼠标绘制污点】", srcImage);
		key = waitKey();
		if (key == 27) //按“Esc ”退出程序
			break;

		//如果检测到键值是1 则恢复为原图  
		if (key == '1')
		{
			g_srcImage.copyTo(srcImage);
			maskImage = Scalar::all(0);
			imshow("【原图-可用鼠标绘制污点】", srcImage);
		}

		//如果检测到回车，则开始执行图像修复  
		Mat dstImage;
		dstImage.create(srcImage.size(), srcImage.type());
		if (key == 13)
		{
			inpaint(srcImage, maskImage, dstImage, 3, INPAINT_TELEA);
			imshow("【修复后的图像】", dstImage);
		}
	}

	return 0;
}