Add File

2025-09-04 14:08:52 +08:00
parent 5c11f4bfe8
commit 6279ba1f95
1 changed files with 124 additions and 0 deletions
--- a/src/main/java/org/dromara/easyai/tools/RgbRegression.java
+++ b/src/main/java/org/dromara/easyai/tools/RgbRegression.java
@@ -0,0 +1,124 @@
+package org.dromara.easyai.tools;
+
+
+import org.dromara.easyai.matrixTools.Matrix;
+import org.dromara.easyai.matrixTools.MatrixOperation;
+
+/**
+ * @param
+ * @DATA
+ * @Author LiDaPeng
+ * @Description rgb回归 Y = r *wr + g * wg + b* wb
+ */
+public class RgbRegression extends MatrixOperation {
+    private float wr;
+    private float wg;
+    private float b;
+    private Matrix RG;//rg矩阵
+    private Matrix B;//b矩阵
+    private Matrix RGB;//rgb矩阵
+    private int xIndex = 0;//记录插入数量
+    private boolean isRegression = false;//是否进行了回归
+    private int regionNub;
+    private int x;
+    private int y;
+
+    public Matrix getRGB() {
+        return RGB;
+    }
+
+    public int getX() {
+        return x;
+    }
+
+    public void setX(int x) {
+        this.x = x;
+    }
+
+    public int getY() {
+        return y;
+    }
+
+    public void setY(int y) {
+        this.y = y;
+    }
+
+    public Matrix getRGMatrix() {
+        return RG;
+    }
+
+    public Matrix getBMatrix() {
+        return B;
+    }
+
+    public void clear(int size) {
+        RG = new Matrix(size, 3);
+        RGB = new Matrix(size, 3);
+        B = new Matrix(size, 1);
+        xIndex = 0;
+        regionNub = size;
+    }
+
+    public int getRegionNub() {
+        return regionNub;
+    }
+
+    public void setRegionNub(int regionNub) {
+        this.regionNub = regionNub;
+    }
+
+    public float getWr() {
+        return wr;
+    }
+
+    public float getWg() {
+        return wg;
+    }
+
+    public float getB() {
+        return b;
+    }
+
+    public RgbRegression(int size) {//初始化rgb矩阵
+        RG = new Matrix(size, 3);
+        RGB = new Matrix(size, 3);
+        B = new Matrix(size, 1);
+        regionNub = size;
+        xIndex = 0;
+    }
+
+    public void insertRGB(float[] rgb) throws Exception {//rgb插入矩阵
+        if (rgb.length == 3) {
+            RGB.setNub(xIndex, 0, rgb[0]);
+            RGB.setNub(xIndex, 1, rgb[1]);
+            RGB.setNub(xIndex, 2, rgb[2]);
+
+            RG.setNub(xIndex, 0, rgb[0]);
+            RG.setNub(xIndex, 1, rgb[1]);
+            RG.setNub(xIndex, 2, 1F);
+
+            B.setNub(xIndex, 0, rgb[2]);
+            xIndex++;
+        } else {
+            throw new Exception("rgb length is not equals three");
+        }
+    }
+
+    public boolean regression() throws Exception {//开始进行回归
+        if (xIndex > 0) {
+            Matrix ws = getLinearRegression(RG, B);
+            if (ws.getX() == 1 && ws.getY() == 1) {//矩阵奇异
+                isRegression = false;
+            } else {
+                wr = ws.getNumber(0, 0);
+                wg = ws.getNumber(1, 0);
+                b = ws.getNumber(2, 0);
+                isRegression = true;
+            }
+            return isRegression;
+            // System.out.println("wr==" + wr + ",wg==" + wg + ",b==" + b);
+        } else {
+            throw new Exception("regression matrix size is zero");
+        }
+    }
+}