C++ OpenCV實現(xiàn)圖像修復(fù)功能

更新時間：2022年01月10日 14:05:01 作者：Zero___Chen

這篇文章主要介紹了通過C++ OpenCV中提供的inpaint API實現(xiàn)對有瑕疵的圖像進(jìn)行修復(fù)，文中的方法講解詳細(xì)，感興趣的同學(xué)可以跟隨小編一起學(xué)習(xí)一下

前言

本文將使用OpenCV C++ 對有瑕疵的圖像進(jìn)行修復(fù)。OpenCV 提供了inpaint API可進(jìn)行圖像修復(fù)。

一、OpenCV inpaint

原圖如圖所示。本案例的需求是希望能夠?qū)D像上的紅線給消除。OpenCV 提供的inpaint API能夠?qū)崿F(xiàn)這個效果。

void inpaint(

InputArray src, 原圖

InputArray inpaintMask, 二進(jìn)制掩模，指示要修復(fù)的像素

OutputArray dst, 目標(biāo)圖像

double inpaintRadius, 像素周圍的鄰域補繪。通常，如果要修復(fù)的區(qū)域很小，則使用較小的值僅產(chǎn)生較少模糊

int flags INPAINT_NS 或 INPAINT_TELEA

)

使用這個API最重要的就是生成一張二進(jìn)制掩模圖像（要修復(fù)的地方）。我們可以使用鼠標(biāo)響應(yīng)事件完成。

	//使用鼠標(biāo)響應(yīng)事件在 inpaintMask 繪制掩模，用于圖片修復(fù)
	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);

		//圖像修復(fù)
		Mat result;
		inpaint(src, inpaintMask, result, 1, INPAINT_NS);
		//imshow("image", result);
	}

如圖為掩模圖像，接下來使用inpaint就可以對圖像進(jìn)行修復(fù)了。

二、源碼

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

/*
void inpaint(
		InputArray src,   原圖
		InputArray inpaintMask, 二進(jìn)制掩模，指示要修復(fù)的像素
		OutputArray dst,  目標(biāo)圖像
		double inpaintRadius, 像素周圍的鄰域補繪。通常，如果要修復(fù)的區(qū)域很小，則使用較小的值僅產(chǎn)生較少模糊
		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*)
{
	//使用鼠標(biāo)響應(yīng)事件在 inpaintMask 繪制掩模，用于圖片修復(fù)
	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);

		//圖像修復(fù)
		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);  //二進(jìn)制掩模，指示要修復(fù)的像素
	setMouseCallback("image", onMouse, NULL);

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