C++ OpenCV图像修复

时间:2022-05-23 Zero___Chen 人气:0

前言

本文将使用OpenCV C++ 对有瑕疵的图像进行修复。OpenCV 提供了inpaint API可进行图像修复。

一、OpenCV inpaint

原图如图所示。本案例的需求是希望能够将图像上的红线给消除。OpenCV 提供的inpaint API能够实现这个效果。

void inpaint(

InputArray src, 原图

InputArray inpaintMask, 二进制掩模，指示要修复的像素

OutputArray dst, 目标图像

double inpaintRadius, 像素周围的邻域补绘。通常，如果要修复的区域很小，则使用较小的值仅产生较少模糊

int flags INPAINT_NS 或 INPAINT_TELEA

)

使用这个API最重要的就是生成一张二进制掩模图像（要修复的地方）。我们可以使用鼠标响应事件完成。

	//使用鼠标响应事件在 inpaintMask 绘制掩模，用于图片修复
	if (event == EVENT_LBUTTONUP || !(flags & EVENT_FLAG_LBUTTON))
	{
		prevPt = Point(-1, -1);
	}
	else if (event == EVENT_LBUTTONDOWN)
	{
		prevPt = Point(x, y);
	}
	else if (event == EVENT_MOUSEMOVE && (flags & EVENT_FLAG_LBUTTON))
	{
		prevPt = Point(x, y);
		//绘制掩模图像
		circle(inpaintMask, prevPt, 2, Scalar::all(255), -1);
		//imshow("inpaintMask", inpaintMask);

		//图像修复
		Mat result;
		inpaint(src, inpaintMask, result, 1, INPAINT_NS);
		//imshow("image", result);
	}

如图为掩模图像，接下来使用inpaint就可以对图像进行修复了。

二、源码

#include <iostream>
#include<opencv2/opencv.hpp>
#include<opencv2/photo.hpp>
using namespace cv;
using namespace std;

/*
void inpaint(
		InputArray src,   原图
		InputArray inpaintMask, 二进制掩模，指示要修复的像素
		OutputArray dst,  目标图像
		double inpaintRadius, 像素周围的邻域补绘。通常，如果要修复的区域很小，则使用较小的值仅产生较少模糊
		int flags     INPAINT_NS  或 INPAINT_TELEA
		)
*/


Mat src, inpaintMask;
Point prevPt(-1, -1);

static void onMouse(int event, int x, int y, int flags, void*)
{
	//使用鼠标响应事件在 inpaintMask 绘制掩模，用于图片修复
	if (event == EVENT_LBUTTONUP || !(flags & EVENT_FLAG_LBUTTON))
	{
		prevPt = Point(-1, -1);
	}
	else if (event == EVENT_LBUTTONDOWN)
	{
		prevPt = Point(x, y);
	}
	else if (event == EVENT_MOUSEMOVE && (flags & EVENT_FLAG_LBUTTON))
	{
		prevPt = Point(x, y);

		//绘制掩模图像
		circle(inpaintMask, prevPt, 2, Scalar::all(255), -1);
		//imshow("inpaintMask", inpaintMask);

		//图像修复
		Mat result;
		inpaint(src, inpaintMask, result, 1, INPAINT_NS);
		//imshow("image", result);
	}
}


int main()
{
	src = imread("girl.jpg");
	
	if (src.empty())
	{
		cout << "No Image!" << endl;
		system("pause");
		return -1;
	}

	namedWindow("image", WINDOW_AUTOSIZE);
	imshow("image", src);

	inpaintMask = Mat::zeros(src.size(), CV_8U);  //二进制掩模，指示要修复的像素
	setMouseCallback("image", onMouse, NULL);

	waitKey(0);
	destroyAllWindows();
	system("pause");
	return 0;	
}

三、效果显示

如图为使用inpaint修复的效果，可以看出基本上将瑕疵都给去掉了。

加载全部内容