8.6 图像修补
基本思想:
利用已经被破坏区域的边缘,即边缘的颜色和结构,繁殖和混合到损坏的图像中,达到图像修补的目的。
8.6.1 实现图像修补:inpaint()函数
1.作用:
用来从扫描的照片中清除灰尘和划痕,从静态图像或视频中去除不需要的物体
2.函数原型:
void inpaint(InputArray src, InputArray inpaintMask, OutputArray dst, double inpaintRadius, int flags)
3.参数说明:
(1)输入图像,8位单通道或三通道
(2)修复掩模,8位单通道,其中的非零像素表示需要修补区域
(3)修补后图像
(4)需要修补的每个点的圆形区域,为修复算法的参考半径
(5)修补方法的标识符,可取值:
1)INPAINT_NS:基于Navier-Stokes方程的方法
2)INPAINT_TELEA:Alexandru Telea方法
8.6.2 综合示例
/*
程序说明:鼠标绘制白色线条破坏原图像图像
键盘按键【1】启动进行图像修复
按键【2】恢复原始图像
*/
#include<opencv2/opencv.hpp>
#include<opencv2/highgui/highgui.hpp>
#include<opencv2/imgproc/imgproc.hpp>
#include<iostream>
using namespace cv;
using namespace std;
//定义辅助宏
#define WINDOW_NAME1 "【原始图窗口】"
#define WINDOW_NAME2 "【修复效果图】"
//全局变量
Mat g_srcImage, g_inpaintMask;
Point previousPoint(-1, -1); //原来的点坐标
//全局函数
static void on_Mouse(int event, int x, int y, int flags, void*);
static void ShowHelpText();
int main()
{
//显示帮助文字
ShowHelpText();
//载入原图并显示
Mat srcImage = imread("Night.jpg");
if (!srcImage.data)
{
printf("载入原图失败~!\n");
return false;
}
imshow(WINDOW_NAME1, srcImage);
//初始化掩模
g_srcImage = srcImage.clone();
g_inpaintMask = Mat::zeros(g_srcImage.size(), CV_8U);
//设置鼠标回调消息
setMouseCallback(WINDOW_NAME1, on_Mouse, 0);
//按键轮询
while (1)
{
//获取按键键值
char c = (char)waitKey();
//键值为ESC,程序退出
if (c == 27) break;
//键值为2,恢复原始图像
if (c == '2')
{
g_inpaintMask = Scalar::all(0);
srcImage.copyTo(g_srcImage);
imshow(WINDOW_NAME1, g_srcImage);
}
//键值为1,进行图像修复
if (c == '1')
{
Mat inpaintedImage;
inpaint(g_srcImage, g_inpaintMask, inpaintedImage, 3, INPAINT_TELEA);
imshow(WINDOW_NAME2, inpaintedImage);
}
}
return 0;
}
//鼠标回调函数
static void on_Mouse(int event, int x, int y, int flags, void*)
{
//鼠标左键弹起消息
if (event == EVENT_LBUTTONUP || !(flags & EVENT_FLAG_LBUTTON))
{
previousPoint = Point(-1, -1);
}
//鼠标左键按下消息
else if (event == EVENT_LBUTTONDOWN)
{
previousPoint = Point(x, y);
}
//鼠标按下状态并移动,进行绘制
else if (event == EVENT_MOUSEMOVE && (flags & EVENT_FLAG_LBUTTON))
{
Point pt(x, y);
if (previousPoint.x < 0)
{
previousPoint = pt;
}
line(g_inpaintMask, previousPoint, pt, Scalar::all(255), 5, 8, 0);
line(g_srcImage, previousPoint, pt, Scalar::all(255), 5, 8, 0);
previousPoint = pt;
imshow(WINDOW_NAME1, g_srcImage);
}
}
static void ShowHelpText()
{
printf("\n\t欢迎来到【图像修复】示例程序~\n");
printf("\n\t请再进行图像修复操作之前,在【原始图】窗口中进行适量的绘制\n");
printf("\n\t按键操作说明:\n");
printf("\t\t\t键盘按键【1】--进行图像修复\n");
printf("\t\t\t键盘按键【2】--恢复原始图\n");
printf("\t\t\t键盘按键【ESC】--退出程序\n");
}
运行效果: