From 551cad3e52dceb09f559838990f32ca90de05d82 Mon Sep 17 00:00:00 2001 From: inter Date: Thu, 4 Sep 2025 14:09:15 +0800 Subject: [PATCH] Add File --- .../org/dromara/easyai/yolo/PositionBack.java | 87 +++++++++++++++++++ 1 file changed, 87 insertions(+) create mode 100644 src/main/java/org/dromara/easyai/yolo/PositionBack.java diff --git a/src/main/java/org/dromara/easyai/yolo/PositionBack.java b/src/main/java/org/dromara/easyai/yolo/PositionBack.java new file mode 100644 index 0000000..7bdd6e2 --- /dev/null +++ b/src/main/java/org/dromara/easyai/yolo/PositionBack.java @@ -0,0 +1,87 @@ +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 PositionBack implements OutBack { + private float distX; + private float distY; + private float width; + private float height; + private float trust; + private Matrix featureMatrix; + + public Matrix getFeatureMatrix() { + return featureMatrix; + } + + public float getTrust() { + return trust; + } + + public float getWidth() { + return width; + } + + public float getHeight() { + return height; + } + + public float getDistX() { + return distX; + } + + public float getDistY() { + return distY; + } + + @Override + public void getBack(float out, int id, long eventId) { + switch (id) { + case 1: + distX = out; + break; + case 2: + distY = out; + break; + case 3: + width = out; + break; + case 4: + height = out; + break; + case 5: + trust = out; + break; + } + } + + @Override + public void getSoftMaxBack(long eventId, List softMax) { + + } + + + @Override + public void backWord(String word, long eventId) { + + } + + @Override + public void getBackMatrix(Matrix matrix, int id, long eventId) { + featureMatrix = matrix; + } + + @Override + public void getWordVector(int id, float w) { + + } + + @Override + public void getBackThreeChannelMatrix(ThreeChannelMatrix picture) { + + } +}