OpenCV学习笔记(十七):图像修补:inpaint()
inpaint()函数
使用区域邻域在图像中还原选定区域。
void inpaint(
InputArray src, // 表示要修复的图像,8位三通道或三通道
InputArray inpaintMask, // 表示修复模板(掩模),8位单通道,非零区域为修补区域
OutputArray dst, // 表示修复后的图像
double inpaintRadius, // 表示该算法所考虑的需要修补的每个点的圆形邻域的圆形领域半径。
int flags // 表示修复使用的算法 // CV_INPAINT_TELEA Alexandru Telea 算法 .// CV_INPAINT_NS 基于 Navier-Stokes 方程的算法
);
感觉两种算法修复效果都还不错,但是都需要事先准备修复模板mask,也就是inpaintMask 这个参数。
例子里面用鼠标在图片上划线,划线的同时也更新了mask,而真正应用的时候需要事先设计好这个mask。
示例:
#include <opencv2/opencv.hpp>using namespace cv;
using namespace std;#define WINDOW_NAME0 "【原始图参考】" //为窗口标题定义的宏
#define WINDOW_NAME1 "【原始图】" //为窗口标题定义的宏
#define WINDOW_NAME2 "【修补后的效果图】" //为窗口标题定义的宏 Mat srcImage0,srcImage1, inpaintMask;
Point previousPoint(-1,-1);//原来的点坐标int main()
{//输出一些帮助信息printf("\n\n\n\t欢迎来到【图像修复】示例程序~\n");printf( "\n\t请在进行图像修复操作之前,在【原始图】窗口中进行适量的绘制""\n\n\t按键操作说明: \n\n""\t\t【鼠标左键】-在图像上绘制白色线条\n\n""\t\t键盘按键【ESC】- 退出程序\n\n""\t\t键盘按键【1】或【SPACE】-进行图像修复操作 \n\n" );//1、载入原图并显示,初始化掩膜和灰度图Mat srcImage = imread("F:/C++/2. OPENCV 3.1.0/TEST/inpaint.jpg", 1);if(!srcImage.data ) { printf("读取图片错误,请确定目录下是否有imread函数指定图片存在~! \n"); return false; }srcImage0 = srcImage.clone();srcImage1 = srcImage.clone();inpaintMask = Mat::zeros(srcImage1.size(), CV_8U);//显示原始图参考imshow(WINDOW_NAME0, srcImage0);//显示原始图imshow(WINDOW_NAME1, srcImage1);//设置鼠标回调消息setMouseCallback( WINDOW_NAME1, On_Mouse, 0 );//轮询按键,根据不同的按键进行处理while (1){//获取按键键值char c = (char)waitKey();//键值为ESC,程序退出if( c == 27 )break;//键值为2,恢复成原始图像if( c == '2' ){inpaintMask = Scalar::all(0);srcImage.copyTo(srcImage1);imshow(WINDOW_NAME1, srcImage1);}//键值为1或者空格,进行图像修补操作if( c == '1' || c == ' ' ){Mat inpaintedImage;inpaint(srcImage1, inpaintMask, inpaintedImage, 3, CV_INPAINT_TELEA);imshow(WINDOW_NAME2, inpaintedImage);}}waitKey(0);return 0;
}
static void On_Mouse( int event, int x, int y, int flags, void* )
{//鼠标左键弹起消息if( event == CV_EVENT_LBUTTONUP || !(flags & CV_EVENT_FLAG_LBUTTON) )previousPoint = Point(-1,-1);//鼠标左键按下消息else if( event == CV_EVENT_LBUTTONDOWN )previousPoint = Point(x,y);//鼠标按下并移动,进行绘制else if( event == CV_EVENT_MOUSEMOVE && (flags & CV_EVENT_FLAG_LBUTTON) ){Point pt(x,y);if( previousPoint.x < 0 )previousPoint = pt;//绘制白色线条// 同时在源图和掩模图 中绘制涂鸦;line( inpaintMask, previousPoint, pt, Scalar::all(255), 5, 8, 0 );line( srcImage1, previousPoint, pt, Scalar::all(255), 5, 8, 0 );previousPoint = pt;imshow(WINDOW_NAME1, srcImage1);}
}
结果:
CV_INPAINT_TELEA 算法:
CV_INPAINT_NS 算法: