添加C#实现
This commit is contained in:
ShaoHua
@@ -0,0 +1,193 @@
|
||||
using System.Collections.Generic;
|
||||
using OpenCvSharp;
|
||||
|
||||
namespace Com.Lmc.ShuiYin.Two.Utils
|
||||
{
|
||||
public class WaterMarkDCT
|
||||
{
|
||||
private static readonly double P = 65;
|
||||
|
||||
public static void Embed(string imagePath, string watermarkPath, string outputPath)
|
||||
{
|
||||
Mat originaleImage = Cv2.ImRead(imagePath);
|
||||
List<Mat> allPlanes = new List<Mat>();
|
||||
Cv2.Split(originaleImage, out Mat[] splitPlanes);
|
||||
allPlanes.AddRange(splitPlanes);
|
||||
Mat YMat = allPlanes[0];
|
||||
|
||||
int[][] watermark = ImageToMatrix(watermarkPath);
|
||||
int length = 8;
|
||||
|
||||
for (int i = 0; i < watermark.Length; i++)
|
||||
{
|
||||
for (int j = 0; j < watermark[0].Length; j++)
|
||||
{
|
||||
Mat block = GetImageValue(YMat, i, j, length);
|
||||
|
||||
int x1 = 1, y1 = 2;
|
||||
int x2 = 2, y2 = 1;
|
||||
|
||||
// In Java: double[] a = block.get(x1, y1);
|
||||
// OpenCvSharp Get returns struct/value.
|
||||
// block is CV_32F (float) based on GetImageValue.
|
||||
|
||||
Cv2.Dct(block, block);
|
||||
|
||||
if (watermark[i][j] == 1)
|
||||
{
|
||||
block.Set(x1, y1, (float)P);
|
||||
block.Set(x2, y2, (float)-P);
|
||||
}
|
||||
if (watermark[i][j] == 0)
|
||||
{
|
||||
block.Set(x1, y1, (float)-P);
|
||||
block.Set(x2, y2, (float)P);
|
||||
}
|
||||
|
||||
Cv2.Idct(block, block);
|
||||
|
||||
for (int m = 0; m < length; m++)
|
||||
{
|
||||
for (int t = 0; t < length; t++)
|
||||
{
|
||||
// In Java: double[] e = block.get(m, t);
|
||||
// YMat.put(i * length + m, j * length + t, e);
|
||||
float e = block.At<float>(m, t);
|
||||
// YMat is likely CV_8U from imread, but GetImageValue reads it.
|
||||
// If YMat is CV_8U, we can't put float P (65) directly if it overflows or logic expects float processing.
|
||||
// But Java code: `Mat YMat = allPlanes.get(0);` (CV_8U usually).
|
||||
// `Mat block = getImageValue(YMat...);` creates new Mat CV_32F.
|
||||
// Then `YMat.put` writes back. If YMat is 8U, putting float might be truncated or requires YMat to be 32F.
|
||||
// However, Java OpenCV `put` handles conversion?
|
||||
// Let's assume YMat should be converted to float if we are doing this, OR we cast to byte.
|
||||
// But P=65 is additive? No, it replaces values.
|
||||
// Wait, `block.get(x1, y1)` returns the DCT coefficient.
|
||||
// The DCT is done on 32F block.
|
||||
// Then IDCT.
|
||||
// Then we put back to YMat.
|
||||
// If YMat is 8U, we are putting float values.
|
||||
// In OpenCvSharp: `YMat.Set(row, col, value)`.
|
||||
// If YMat is 8U, Set will cast.
|
||||
YMat.Set(i * length + m, j * length + t, (byte)e); // Cast to byte? Or keep as double/float if YMat was converted?
|
||||
// Java code didn't convert YMat to 32F explicitly in Embed method, unlike DFT.
|
||||
// But `getImageValue` creates a 32F mat from YMat values.
|
||||
// `YMat.put(...)` in Java takes double array.
|
||||
// If YMat is 8U, it sets the byte value.
|
||||
// So casting to byte is correct.
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Mat imageOut = new Mat();
|
||||
Cv2.Merge(allPlanes.ToArray(), imageOut);
|
||||
Cv2.ImWrite(outputPath, imageOut);
|
||||
}
|
||||
|
||||
public static void Extract(string targetImage, string outputWatermark)
|
||||
{
|
||||
Mat image = Cv2.ImRead(targetImage);
|
||||
List<Mat> allPlanes = new List<Mat>();
|
||||
Cv2.Split(image, out Mat[] splitPlanes);
|
||||
allPlanes.AddRange(splitPlanes);
|
||||
Mat YMat = allPlanes[0];
|
||||
|
||||
int rows = 100;
|
||||
int cols = 100;
|
||||
int length = 8;
|
||||
int[][] watermark = new int[rows][];
|
||||
for (int k = 0; k < rows; k++) watermark[k] = new int[cols];
|
||||
|
||||
for (int i = 0; i < rows; i++)
|
||||
{
|
||||
for (int j = 0; j < cols; j++)
|
||||
{
|
||||
Mat block = GetImageValue(YMat, i, j, length);
|
||||
Cv2.Dct(block, block);
|
||||
|
||||
int x1 = 1, y1 = 2;
|
||||
int x2 = 2, y2 = 1;
|
||||
|
||||
float a = block.At<float>(x1, y1);
|
||||
float c = block.At<float>(x2, y2);
|
||||
|
||||
if (a >= c)
|
||||
{
|
||||
watermark[i][j] = 1;
|
||||
}
|
||||
}
|
||||
}
|
||||
MatrixToImage(watermark, outputWatermark);
|
||||
}
|
||||
|
||||
private static Mat GetImageValue(Mat YMat, int x, int y, int length)
|
||||
{
|
||||
Mat mat = new Mat(length, length, MatType.CV_32F);
|
||||
for (int i = 0; i < length; i++)
|
||||
{
|
||||
for (int j = 0; j < length; j++)
|
||||
{
|
||||
// Java: double[] temp = YMat.get(x * length + i, y * length + j);
|
||||
// mat.put(i, j, temp);
|
||||
// If YMat is 8U, At<byte>
|
||||
byte val = YMat.At<byte>(x * length + i, y * length + j);
|
||||
mat.Set(i, j, (float)val);
|
||||
}
|
||||
}
|
||||
return mat;
|
||||
}
|
||||
|
||||
private static void MatrixToImage(int[][] watermark, string dstPath)
|
||||
{
|
||||
int rows = watermark.Length;
|
||||
int columns = watermark[0].Length;
|
||||
Mat image = new Mat(rows, columns, MatType.CV_8U); // THRESH_BINARY usually on 8U
|
||||
for (int i = 0; i < rows; i++)
|
||||
{
|
||||
for (int j = 0; j < columns; j++)
|
||||
{
|
||||
if (watermark[i][j] == 1)
|
||||
{
|
||||
image.Set(i, j, (byte)255);
|
||||
}
|
||||
else
|
||||
{
|
||||
image.Set(i, j, (byte)0);
|
||||
}
|
||||
}
|
||||
}
|
||||
Cv2.ImWrite(dstPath, image);
|
||||
}
|
||||
|
||||
private static int[][] ImageToMatrix(string srcPath)
|
||||
{
|
||||
Mat mat = Cv2.ImRead(srcPath, ImreadModes.Grayscale); // Java: THRESH_BINARY is not a load flag, it's a thresh type. But here passed to imread?
|
||||
// Java: imread(srcPath, Imgproc.THRESH_BINARY).
|
||||
// THRESH_BINARY is 0. CV_LOAD_IMAGE_GRAYSCALE is 0.
|
||||
// So they are loading as grayscale.
|
||||
|
||||
int rows = mat.Rows;
|
||||
int columns = mat.Cols;
|
||||
int[][] waterMark = new int[rows][];
|
||||
for(int k=0; k<rows; k++) waterMark[k] = new int[columns];
|
||||
|
||||
for (int i = 0; i < rows; i++)
|
||||
{
|
||||
for (int j = 0; j < columns; j++)
|
||||
{
|
||||
// double[] doubles = mat.get(i, j);
|
||||
byte val = mat.At<byte>(i, j);
|
||||
if (val == 255)
|
||||
{
|
||||
waterMark[i][j] = 1;
|
||||
}
|
||||
else
|
||||
{
|
||||
waterMark[i][j] = 0;
|
||||
}
|
||||
}
|
||||
}
|
||||
return waterMark;
|
||||
}
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user