简单图像修复inpaint
#include<iostream>
#include<opencv2/opencv.hpp>
using namespace std;
using namespace cv;
Mat img, inpaintmask;
Point prevpt(0, 0);
//点击鼠标绘制噪声
static void onmouse(int event, int x, int y, int flags, void*);
int main()
{
Mat scr;
scr = imread("C:/Users/MECHREVO/Desktop/test.jpg");
if (scr.empty())
{
cout << "无法打开" << endl;
return 0;
}
img = scr.clone();
inpaintmask = Mat::zeros(img.size(), CV_8U);
imshow("img", img);
setMouseCallback("img", onmouse, 0);
//完成图像修复
waitKey(0);//等待操作,单击窗口叉退出鼠标事件
setMouseCallback("img", NULL, NULL);//解除鼠标事件
imshow("img", img);
imshow("原图无污点scr", scr);
Mat remat;
inpaint(img, inpaintmask, remat, 5, CV_INPAINT_TELEA);
imshow("修复后的remat", remat);
waitKey(0);
return 0;
}
static void onmouse(int event, int x, int y, int flags, void*)
{
//鼠标响应事件
if (event == CV_EVENT_LBUTTONUP ||(flags !=CV_EVENT_FLAG_LBUTTON))
prevpt = Point(0, 0);//初始化第一个坐标
//按下左键
if (event == CV_EVENT_LBUTTONDOWN)
prevpt = Point(x, y);
//鼠标移动和左键事件
if (event == CV_EVENT_MOUSEMOVE && (flags == CV_EVENT_FLAG_LBUTTON))
{
Point pt(x, y);
if (prevpt.x < 0)
{
prevpt = pt;
}
//绘制鼠标移动轨迹
line(inpaintmask, prevpt, pt, Scalar(255), 5, 8, 0);
line(img, prevpt, pt, Scalar::all(255), 5, 8, 0);
prevpt = pt;
imshow("img", img);
}
}