C++ OpenCV實現(xiàn)圖像去水印功能

更新時間：2022年01月14日 08:33:42 作者：Zero___Chen

本文將介紹如何使用OpenCV C++ 進行簡單圖像水印去除。我們在網(wǎng)上download圖片時，經(jīng)常因為版權(quán)問題有水印。本案例通過編寫算法進行簡單水印去除。需要的可以參考一下

前言

本文將使用OpenCV C++ 進行簡單圖像水印去除。我們在網(wǎng)上download圖片時，經(jīng)常因為版權(quán)問題有水印。本案例通過編寫算法進行簡單水印去除。

一、水印定位

如圖所示，圖像左下角、右下角有水印。第一步，我們首先得定位水印所在位置。

	Mat gray;
	cvtColor(src, gray, COLOR_BGR2GRAY);

	//圖像二值化，篩選出白色區(qū)域部分
	Mat thresh;
	threshold(gray, thresh, 220, 255, THRESH_BINARY);

如圖為二值化后得到的效果圖。接下來，我們需要提取水印區(qū)域的像素。

//提取圖片下方的水印，制作掩模圖像
?? ?Mat mask = Mat::zeros(src.size(), CV_8U);
?? ?int height = src.rows;
?? ?int width = src.cols;
?? ?int start = 0.9*height;
?? ?//遍歷圖像像素，提取出水印部分像素，制作掩模圖像
?? ?for (int i = start; i < height; i++)
?? ?{
?? ??? ?uchar*data = thresh.ptr<uchar>(i);
?? ??? ?for (int j = 0; j < width; j++)
?? ??? ?{
?? ??? ??? ?if (data[j] == 255)
?? ??? ??? ?{
?? ??? ??? ??? ?mask.at<uchar>(i, j) = 255;?? ??? ??? ?
?? ??? ??? ?}?? ??? ??? ?
?? ??? ?}
?? ?}

如圖所示，至此我們已經(jīng)得到水印部分的二值掩模圖像。接下來，使用OpenCV 內(nèi)置inpaint API進行圖像像素修復(fù)。具體可以參考OpenCV C++案例實戰(zhàn)十四《圖像修復(fù)》

二、圖像修復(fù)

 //使用inpaint進行圖像修復(fù)
    Mat result;
    inpaint(src, mask, result, 1, INPAINT_NS);

如圖所示，這是直接使用提取出的二值掩模進行圖像修復(fù)得到的結(jié)果，可以看出效果不是很好。原因是，提取出來的掩模未能覆蓋完全待修復(fù)像素。故我們需要將掩模圖像進行膨脹操作，擴大掩模范圍。

    //將掩模進行膨脹，使其能夠覆蓋圖像更大區(qū)域
    Mat kernel = getStructuringElement(MORPH_RECT, Size(5, 5));
    dilate(mask, mask, kernel);

如圖為進行膨脹過后的掩模圖像。這時，我們使用這張掩模圖像再進行圖像修復(fù)工作，得到效果如圖所示。

三、效果顯示

四、源碼

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

/*
void inpaint(
?? ??? ?InputArray src, ? 原圖
?? ??? ?InputArray inpaintMask, 二進制掩模，指示要修復(fù)的像素
?? ??? ?OutputArray dst, ?目標圖像
?? ??? ?double inpaintRadius, 像素周圍的鄰域補繪。通常，如果要修復(fù)的區(qū)域很小，則使用較小的值僅產(chǎn)生較少模糊
?? ??? ?int flags ? ? INPAINT_NS ?或 INPAINT_TELEA
?? ??? ?)
*/

/*
圖像簡單水印去除：定位到水印所在位置，將水印部分提取出來制作二進制掩模圖像，使用inpaint API進行圖像修復(fù)
*/

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

?? ?Mat gray;
?? ?cvtColor(src, gray, COLOR_BGR2GRAY);

?? ?//圖像二值化，篩選出白色區(qū)域部分
?? ?Mat thresh;
?? ?threshold(gray, thresh, 220, 255, THRESH_BINARY);

?? ?//提取圖片下方的水印，制作掩模圖像
?? ?Mat mask = Mat::zeros(src.size(), CV_8U);
?? ?int height = src.rows;
?? ?int width = src.cols;
?? ?int start = 0.9*height;
?? ?//遍歷圖像像素，提取出水印部分像素，制作掩模圖像
?? ?for (int i = start; i < height; i++)
?? ?{
?? ??? ?uchar*data = thresh.ptr<uchar>(i);
?? ??? ?for (int j = 0; j < width; j++)
?? ??? ?{
?? ??? ??? ?if (data[j] == 255)
?? ??? ??? ?{
?? ??? ??? ??? ?mask.at<uchar>(i, j) = 255;?? ??? ??? ?
?? ??? ??? ?}?? ??? ??? ?
?? ??? ?}
?? ?}

?? ?//將掩模進行膨脹，使其能夠覆蓋圖像更大區(qū)域
?? ?Mat kernel = getStructuringElement(MORPH_RECT, Size(5, 5));
?? ?dilate(mask, mask, kernel);

?? ?//使用inpaint進行圖像修復(fù)
?? ?Mat result;
?? ?inpaint(src, mask, result, 1, INPAINT_NS);

?? ?imshow("mask", mask);
?? ?imshow("test", result);
?? ?waitKey(0);
?? ?system("pause");
?? ?return 0;
}