java 盲水印_Java使用OpenCV 基于离散傅里叶变换算法 实现图片盲水印添加

import org.opencv.core.*;importorg.opencv.imgproc.Imgproc;importjava.util.ArrayList;importjava.util.List;/***@authoryangxiaohui

* @Date: Create by 2018-10-25 19:14

* @Description: 添加图片盲水印工具类*/

public classImgWatermarkUtil {private static List planes = new ArrayList();private static List allPlanes = new ArrayList();/***

* 添加图片文字水印

*

*@authorYangxiaohui

* @date 2018-10-25 19:16

*@paramimage 图片对象

*@paramwatermarkText 水印文字*/

public staticMat addImageWatermarkWithText(Mat image, String watermarkText){

Mat complexImage= newMat();//优化图像的尺寸//Mat padded = optimizeImageDim(image);

Mat padded =splitSrc(image);

padded.convertTo(padded, CvType.CV_32F);

planes.add(padded);

planes.add(Mat.zeros(padded.size(), CvType.CV_32F));

Core.merge(planes, complexImage);//dft

Core.dft(complexImage, complexImage);//添加文本水印

Scalar scalar = new Scalar(0, 0, 0);

Point point= new Point(40, 40);

Core.putText(complexImage, watermarkText, point, Core.FONT_HERSHEY_DUPLEX, 1D, scalar);

Core.flip(complexImage, complexImage,-1);

Core.putText(complexImage, watermarkText, point, Core.FONT_HERSHEY_DUPLEX, 1D, scalar);

Core.flip(complexImage, complexImage,-1);returnantitransformImage(complexImage, allPlanes);

}/***

* 获取图片水印

*

*@authorYangxiaohui

* @date 2018-10-25 19:58

*@paramimage*/

public staticMat getImageWatermarkWithText(Mat image){

List planes = new ArrayList();

Mat complexImage= newMat();

Mat padded=splitSrc(image);

padded.convertTo(padded, CvType.CV_32F);

planes.add(padded);

planes.add(Mat.zeros(padded.size(), CvType.CV_32F));

Core.merge(planes, complexImage);//dft

Core.dft(complexImage, complexImage);

Mat magnitude=createOptimizedMagnitude(complexImage);

planes.clear();returnmagnitude;

}private staticMat splitSrc(Mat mat) {

mat=optimizeImageDim(mat);

Core.split(mat, allPlanes);

Mat padded= newMat();if (allPlanes.size() > 1) {for (int i = 0; i < allPlanes.size(); i++) {if (i == 0) {

padded=allPlanes.get(i);break;

}

}

}else{

padded=mat;

}returnpadded;

}private static Mat antitransformImage(Mat complexImage, ListallPlanes) {

Mat invDFT= newMat();

Core.idft(complexImage, invDFT, Core.DFT_SCALE| Core.DFT_REAL_OUTPUT, 0);

Mat restoredImage= newMat();

invDFT.convertTo(restoredImage, CvType.CV_8U);if (allPlanes.size() == 0) {

allPlanes.add(restoredImage);

}else{

allPlanes.set(0, restoredImage);

}

Mat lastImage= newMat();

Core.merge(allPlanes, lastImage);returnlastImage;

}/***

* 为加快傅里叶变换的速度，对要处理的图片尺寸进行优化

*

*@authorYangxiaohui

* @date 2018-10-25 19:33

*@paramimage

*@return

*/

private staticMat optimizeImageDim(Mat image) {

Mat padded= newMat();int addPixelRows =Core.getOptimalDFTSize(image.rows());int addPixelCols =Core.getOptimalDFTSize(image.cols());

Imgproc.copyMakeBorder(image, padded,0, addPixelRows - image.rows(), 0, addPixelCols -image.cols(),

Imgproc.BORDER_CONSTANT, Scalar.all(0));returnpadded;

}private staticMat createOptimizedMagnitude(Mat complexImage) {

List newPlanes = new ArrayList();

Mat mag= newMat();

Core.split(complexImage, newPlanes);

Core.magnitude(newPlanes.get(0), newPlanes.get(1), mag);

Core.add(Mat.ones(mag.size(), CvType.CV_32F), mag, mag);

Core.log(mag, mag);

shiftDFT(mag);

mag.convertTo(mag, CvType.CV_8UC1);

Core.normalize(mag, mag,0, 255, Core.NORM_MINMAX, CvType.CV_8UC1);returnmag;

}private static voidshiftDFT(Mat image) {

image= image.submat(new Rect(0, 0, image.cols() & -2, image.rows() & -2));int cx = image.cols() / 2;int cy = image.rows() / 2;

Mat q0= new Mat(image, new Rect(0, 0, cx, cy));

Mat q1= new Mat(image, new Rect(cx, 0, cx, cy));

Mat q2= new Mat(image, new Rect(0, cy, cx, cy));

Mat q3= new Mat(image, newRect(cx, cy, cx, cy));

Mat tmp= newMat();

q0.copyTo(tmp);

q3.copyTo(q0);

tmp.copyTo(q3);

q1.copyTo(tmp);

q2.copyTo(q1);

tmp.copyTo(q2);

}

}