1、FaceNet 特征提取新增人脸对齐

2、人脸检测新5点人脸关键点定位
3、特征提取接口支持多人脸和最佳人脸提取
4、修复人脸框边界精度问题
5、更新 Maven 发布的 groupId

smartjavaai-common/src/main/java/cn/smartjavaai/common/utils/OpenCVUtils.java

@@ -0,0 +1,119 @@
+package cn.smartjavaai.common.utils;
+
+import ai.djl.ndarray.NDArray;
+import ai.djl.ndarray.NDManager;
+import org.opencv.core.CvType;
+import org.opencv.core.Mat;
+import org.opencv.core.Point;
+import org.opencv.core.Scalar;
+import org.opencv.imgproc.Imgproc;
+
+import java.awt.image.BufferedImage;
+import java.awt.image.DataBufferByte;
+
+/**
+ * OpenCV 工具类
+ */
+public class OpenCVUtils {
+    /**
+     * canny算法，边缘检测
+     *
+     * @param src
+     * @return
+     */
+    public static Mat canny(Mat src) {
+        Mat mat = src.clone();
+        Imgproc.Canny(src, mat, 100, 200);
+        return mat;
+    }
+
+    /**
+     * 画线
+     *
+     * @param mat
+     * @param point1
+     * @param point2
+     */
+    public static void line(Mat mat, Point point1, Point point2) {
+        Imgproc.line(mat, point1, point2, new Scalar(255, 255, 255), 1);
+    }
+
+    /**
+     * NDArray to opencv_core.Mat
+     *
+     * @param manager
+     * @param srcPoints
+     * @param dstPoints
+     * @return
+     */
+    public static Mat toOpenCVMat(NDManager manager, NDArray srcPoints, NDArray dstPoints) {
+        NDArray svdMat = SVDUtils.transformationFromPoints(manager, srcPoints, dstPoints);
+        double[] doubleArray = svdMat.toDoubleArray();
+        Mat newSvdMat = new Mat(2, 3, CvType.CV_64F);
+        for (int i = 0; i < 2; i++) {
+            for (int j = 0; j < 3; j++) {
+                newSvdMat.put(i, j, doubleArray[i * 3 + j]);
+            }
+        }
+        return newSvdMat;
+    }
+
+    /**
+     * double[][] points array to Mat
+     * @param points
+     * @return
+     */
+    public static Mat toOpenCVMat(double[][] points) {
+        Mat mat = new Mat(5, 2, CvType.CV_64F);
+        for (int i = 0; i < 5; i++) {
+            for (int j = 0; j < 2; j++) {
+                mat.put(i, j, points[i * 5 + j]);
+            }
+        }
+        return mat;
+    }
+
+    /**
+     * 变换矩阵的逆矩阵
+     *
+     * @param src
+     * @return
+     */
+    public static Mat invertAffineTransform(Mat src) {
+        Mat dst = src.clone();
+        Imgproc.invertAffineTransform(src, dst);
+        return dst;
+    }
+
+    /**
+     * Mat to BufferedImage
+     *
+     * @param mat
+     * @return
+     */
+    public static BufferedImage mat2Image(Mat mat) {
+        int width = mat.width();
+        int height = mat.height();
+        byte[] data = new byte[width * height * (int) mat.elemSize()];
+        Imgproc.cvtColor(mat, mat, 4);
+        mat.get(0, 0, data);
+        BufferedImage ret = new BufferedImage(width, height, 5);
+        ret.getRaster().setDataElements(0, 0, width, height, data);
+        return ret;
+    }
+
+    /**
+     * BufferedImage to Mat
+     *
+     * @param img
+     * @return
+     */
+    public static Mat image2Mat(BufferedImage img) {
+        int width = img.getWidth();
+        int height = img.getHeight();
+        byte[] data = ((DataBufferByte) img.getRaster().getDataBuffer()).getData();
+        Mat mat = new Mat(height, width, CvType.CV_8UC3);
+        mat.put(0, 0, data);
+        return mat;
+    }
+}