From 6cdcd80e245ba21584da4c19ee845e9498d86f55 Mon Sep 17 00:00:00 2001 From: inter Date: Thu, 4 Sep 2025 14:09:14 +0800 Subject: [PATCH] Add File --- .../org/dromara/easyai/yolo/YoloTypeBack.java | 72 +++++++++++++++++++ 1 file changed, 72 insertions(+) create mode 100644 src/main/java/org/dromara/easyai/yolo/YoloTypeBack.java diff --git a/src/main/java/org/dromara/easyai/yolo/YoloTypeBack.java b/src/main/java/org/dromara/easyai/yolo/YoloTypeBack.java new file mode 100644 index 0000000..cde845c --- /dev/null +++ b/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 softMax; + + public void setOut(float out) { + this.out = out; + } + + public void setId(int id) { + this.id = id; + } + + public float getOut() { + return out; + } + + public List 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 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) { + + } +}