集成算法seetaface6

smartjavaai-common/pom.xml

@@ -6,7 +6,7 @@
     <parent>
         <groupId>ink.numberone</groupId>
         <artifactId>smartjavaai-parent</artifactId>
-        <version>1.0.4</version>
+        <version>1.0.5</version>
     </parent>
 
     <artifactId>smartjavaai-common</artifactId>

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

@@ -0,0 +1,77 @@
+package cn.smartjavaai.common.utils;
+
+import java.awt.image.BufferedImage;
+//import java.awt.image.ColorConvertOp;
+import java.awt.image.ComponentSampleModel;
+import java.util.Arrays;
+
+/**
+ * 图片处理工具类
+ */
+public class ImageUtils {
+    /**
+     * @param image
+     * @param bandOffset 用于推断通道顺序
+     * @return
+     */
+    private static boolean equalBandOffsetWith3Byte(BufferedImage image, int[] bandOffset) {
+        if (image.getType() == BufferedImage.TYPE_3BYTE_BGR) {
+            if (image.getData().getSampleModel() instanceof ComponentSampleModel) {
+                ComponentSampleModel sampleModel = (ComponentSampleModel) image.getData().getSampleModel();
+                if (Arrays.equals(sampleModel.getBandOffsets(), bandOffset)) {
+                    return true;
+                }
+            }
+        }
+        return false;
+    }
+
+    /**
+     * 推断图像是否为BGR格式
+     *
+     * @return
+     */
+    public static boolean isBGR3Byte(BufferedImage image) {
+        return equalBandOffsetWith3Byte(image, new int[]{0, 1, 2});
+    }
+
+    /**
+     * 对图像解码返回BGR格式矩阵数据
+     *
+     * @param image
+     * @return
+     */
+    public static byte[] getMatrixBGR(BufferedImage image) {
+        byte[] matrixBGR;
+        if (isBGR3Byte(image)) {
+            matrixBGR = (byte[]) image.getData().getDataElements(0, 0, image.getWidth(), image.getHeight(), null);
+        } else {
+            // ARGB格式图像数据
+            int intrgb[] = image.getRGB(0, 0, image.getWidth(), image.getHeight(), null, 0, image.getWidth());
+            matrixBGR = new byte[image.getWidth() * image.getHeight() * 3];
+            // ARGB转BGR格式
+            for (int i = 0, j = 0; i < intrgb.length; ++i, j += 3) {
+                matrixBGR[j] = (byte) (intrgb[i] & 0xff);
+                matrixBGR[j + 1] = (byte) ((intrgb[i] >> 8) & 0xff);
+                matrixBGR[j + 2] = (byte) ((intrgb[i] >> 16) & 0xff);
+            }
+        }
+        return matrixBGR;
+    }
+
+
+    public static BufferedImage bgrToBufferedImage(byte[] data, int width, int height) {
+        int type = BufferedImage.TYPE_3BYTE_BGR;
+        // bgr to rgb
+        byte b;
+        for (int i = 0; i < data.length; i = i + 3) {
+            b = data[i];
+            data[i] = data[i + 2];
+            data[i + 2] = b;
+        }
+        BufferedImage image = new BufferedImage(width, height, type);
+        image.getRaster().setDataElements(0, 0, width, height, data);
+        return image;
+    }
+
+}