javacv实现图片去水印

最近工作需求要做图片去水印的功能,研究了一段时间发现javacv能做这个功能,然后总结了一下。
首先导入maven


            org.bytedeco
            javacv
            1.4.3
            
                
                    org.bytedeco
                    javacpp
                
                
                    org.bytedeco.javacpp-presets
                    flycapture
                
                
                    org.bytedeco.javacpp-presets
                    libdc1394
                
                
                    org.bytedeco.javacpp-presets
                    libfreenect
                
                
                    org.bytedeco.javacpp-presets
                    libfreenect2
                
                
                    org.bytedeco.javacpp-presets
                    librealsense
                
                
                    org.bytedeco.javacpp-presets
                    videoinput
                
                
                    org.bytedeco.javacpp-presets
                    tesseract
                
                
                    org.bytedeco.javacpp-presets
                    leptonica
                
                
                    org.bytedeco.javacpp-presets
                    flandmark
                
                
                    org.bytedeco.javacpp-presets
                    artoolkitplus
                
            
        

            org.bytedeco
            javacv-platform
            1.4.3
            
                
                    org.bytedeco
                    javacv
                
                
                    org.bytedeco.javacpp-presets
                    flycapture-platform
                
                
                    org.bytedeco.javacpp-presets
                    libdc1394-platform
                
                
                    org.bytedeco.javacpp-presets
                    libfreenect-platform
                
                
                    org.bytedeco.javacpp-presets
                    libfreenect2-platform
                
                
                    org.bytedeco.javacpp-presets
                    librealsense-platform
                
                
                    org.bytedeco.javacpp-presets
                    videoinput-platform
                
                
                    org.bytedeco.javacpp-presets
                    tesseract-platform
                
                
                    org.bytedeco.javacpp-presets
                    leptonica-platform
                
                
                    org.bytedeco.javacpp-presets
                    flandmark-platform
                
                
                    org.bytedeco.javacpp-presets
                    artoolkitplus-platform
                
            
        
        
            org.bytedeco.javacpp-presets
            ffmpeg
            4.0.2-1.4.3
        

开始进行实现:
第一步:
前景背景求差法

#include 
#include 
#include 
#include "opencv2/photo/photo.hpp"
 
#include 
 
using namespace std;
using namespace cv;
 
int main(int argc, char** argv)
{
	cv::VideoCapture cap("C:/Users/zyy/Desktop/test_logs/test.mp4");
	if (!cap.isOpened())
	{
		cout << "Failed!" << endl;
		return -1;
	}
 
	Mat result, mask, dst1;
	Mat inpaintMask1;
	Mat inpaintMask = imread("C:/Users/zyy/Desktop/test_logs/log5.jpg");
	cv::cvtColor(inpaintMask, inpaintMask, CV_BGR2GRAY);
	cv::namedWindow("Result", 1);
	cv::namedWindow("Frame", 1);
	int i = 0;
	for (;;)
	{
		Mat frame;
		cap >> frame;
		if (!frame.data)
		{
			cout << "Over!" << endl;
			break;
		}
 
		if (i <= 2)  //看视频,自己调试确定这个值,二十几帧之后左上角那个小图才出来,所以选了30
		{
			if (i == 2)
			{
				mask = frame;
				cv::cvtColor(mask, mask, CV_BGR2GRAY);
				//Mat inpaintMask1(mask.rows, mask.cols, CV_32FC1, 1);
				inpaintMask1 = mask;
			}
		}
		else
		{
			cv::bitwise_and(result, mask, mask);//前景背景求差法
			if (i > 2 && i <= 40)
			{
				cv::cvtColor(frame, result, CV_BGR2GRAY);
				cv::bitwise_and(result, mask, mask);//前景背景求差法
				inpaintMask1 = mask;
				imwrite("C:/Users/zyy/Desktop/test_logs/www.jpg", inpaintMask1);
			}
			Mat inpainted;
			inpaint(frame, inpaintMask, inpainted, 8, CV_INPAINT_TELEA);
			imshow("inpainted image", inpainted);
 
			imshow("Frame", frame);
			imshow("Result", mask);
			//waitKey();
			if (cv::waitKey(33) >= 0) break;
		}
		i++;
	}
 
	waitKey();
	return 0;
}

第二步:
logo和水印位置检测

#include 
#include "cv.h"
#include "cxcore.h"
#include "highgui.h"
using namespace std;
int main()
{
	IplImage *src = cvLoadImage("C:/Users/zyy/Desktop/test_logs/22.jpg", 0);
	IplImage *srcResult = cvLoadImage("C:/Users/zyy/Desktop/test_logs/22.jpg", 3);  //用来显示
	IplImage *templat = cvLoadImage("C:/Users/zyy/Desktop/test_logs/yk2.jpg", 0);
	IplImage *result;
	if (!src || !templat)
	{
		cout << "打开图像失败" << endl;
		return 0;
	}
	int srcW, srcH, templatW, templatH, resultH, resultW;
	srcW = src->width;
	srcH = src->height;
	templatW = templat->width;
	templatH = templat->height;
	if (srcW < templatW || srcH < templatH)
	{
		cout << "模板不能比原图像小" << endl;
		return 0;
	}
	resultW = srcW - templatW + 1;
	resultH = srcH - templatH + 1;
	result = cvCreateImage(cvSize(resultW, resultH), 32, 1);
	cvMatchTemplate(src, templat, result, CV_TM_SQDIFF_NORMED);//CV_TM_SQDIFF_NORMED
	double minValue, maxValue;
	CvPoint minLoc, maxLoc;
	cvMinMaxLoc(result, &minValue, &maxValue, &minLoc, &maxLoc);
	cout << "最小值:" << minValue << " 最大值:" << maxValue << endl;
	cvRectangle(srcResult, minLoc, cvPoint(minLoc.x + templatW, minLoc.y + templatH), cvScalar(0, 0, 255));
	cvNamedWindow("srcResult", 0);
	cvNamedWindow("templat", 0);
	cvShowImage("srcResult", srcResult);
	cvShowImage("templat", templat);
	cvWaitKey(0);
	cvReleaseImage(&result);
	cvReleaseImage(&templat);
	cvReleaseImage(&srcResult);
	cvReleaseImage(&src);
	return 0;
}

第三步:
在已经求得掩膜图像的情况下,进行去logo操作。

    static String video_path="D://video//videos//";
    public static void ets( ) {
        Mat inpaintMask = new Mat();
        Mat inpainted=new Mat();
        Mat template = imread("D://video//movie_10.jpg");
        Mat frame = new Mat();
        cvtColor(template, inpaintMask, COLOR_BGR2GRAY);
        VideoCapture cap = new VideoCapture("D://video//mp4//1553583032966-480p.mp4");
        if(!cap.isOpened())
            return;
        String fileName=video_path+"VideoTest12.mp4";
        Size size = new Size((int)cap.get(CV_CAP_PROP_FRAME_WIDTH), (int)cap.get(CV_CAP_PROP_FRAME_HEIGHT));
        double fps = cap.get(CV_CAP_PROP_FPS);
        VideoWriter vw1 = new VideoWriter(fileName, CV_FOURCC((byte)'D',(byte)'I',(byte)'V',(byte)'3'), fps/6, size, true);

        int start =11;
        int end =20;
        for (int i=start;i

上述的图像和MP4文件自己准备就可以。感兴趣的可以去测试了。

你可能感兴趣的:(javacv,java)