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

图像修复技术就是利用图像中损坏区域边缘的像素，根据像素值的大小以及像素间的结构关系，估计出损坏区域可能的像素排列，从而去除图像中受污染的区域。图像修复不仅可以去除图像中得“划痕”，还可以去除图像中得水印、日期等。

OpenCV 提供了能够对含有较少污染或者水印的图像进行修复的函数inpaint()，其原型如下:

void cv::inpaint(	InputArray 	src,
					InputArray 	inpaintMask,
					OutputArray 	dst,
					double 	inpaintRadius,
					int 	flags
				)

参数意义如下：

src---待修复图像，在OpenCV3中，可以为单通道或三通道图像，但数据类型只能为CV_8U。在OpenCV4中，也可以为单通道或三通道图像，当图像为单通道时，数据类型可以是CV_8U、CV_16U或者CV_32F，当图像为三通道时数据类型必须是CV_8U。

inpaintMask---掩码图像。数据类型为CV_8U的单通道图像，与待修复图像具有相同的尺寸。非零元素对应位置的像素表示需要进行修复处理。

dst---输出图像，与输入图像具有相同的大小和数据类型。

inpaintRadius--修复算法考虑的每个像素点的圆形邻域半径。

flags---选择使用何种修复算法。

标志参数	简记	含义
INPAINT_NS	0	基于Navier-Stokes算法修复图像
INPAINT_TELEA	1	基于Alexandru Telea算法修复图像

示例代码如下：

代码中用到的图像下载链接：链接：https://pan.baidu.com/s/13gKSTVRWbpkHvrTMSYtwBg?pwd=po9z 

//博主微信/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;
}

这里就不对代码进行介绍了，注释已经对程序说明得比较清楚了。

关于窗口鼠标事件的设置，可以参看我的另一篇博文，链接如下：

https://blog.csdn.net/wenhao_ir/article/details/51538445

运行结果如下图所示：

 从运行结果中可以看出，修复效果还是不错的。