Add File

src/main/java/org/dromara/easyai/yolo/YoloTypeBack.java

@@ -0,0 +1,72 @@
+package org.dromara.easyai.yolo;
+
+import org.dromara.easyai.entity.ThreeChannelMatrix;
+import org.dromara.easyai.matrixTools.Matrix;
+import org.dromara.easyai.i.OutBack;
+
+import java.util.List;
+
+public class YoloTypeBack implements OutBack {
+    private float out = -1;
+    private int id = 0;
+    private List<Float> softMax;
+
+    public void setOut(float out) {
+        this.out = out;
+    }
+
+    public void setId(int id) {
+        this.id = id;
+    }
+
+    public float getOut() {
+        return out;
+    }
+
+    public List<Float> getSoftMax() {
+        return softMax;
+    }
+
+    public int getId() {
+        return id;
+    }
+
+    public void clear() {
+        out = -1;
+        id = 0;
+    }
+
+    @Override
+    public void getBack(float out, int id, long eventId) {
+        if (out > this.out) {
+            this.out = out;
+            this.id = id;
+        }
+    }
+
+    @Override
+    public void getSoftMaxBack(long eventId, List<Float> softMax) {
+        this.softMax = softMax;
+    }
+
+
+    @Override
+    public void backWord(String word, long eventId) {
+
+    }
+
+    @Override
+    public void getBackMatrix(Matrix matrix, int id, long eventId) {
+
+    }
+
+    @Override
+    public void getWordVector(int id, float w) {
+
+    }
+
+    @Override
+    public void getBackThreeChannelMatrix(ThreeChannelMatrix picture) {
+
+    }
+}