From ffce5a043ec4d92a2ac9b7cfe0a832857d7303e3 Mon Sep 17 00:00:00 2001
From: inter <inter@qq.com>
Date: Thu, 4 Sep 2025 14:09:35 +0800
Subject: [PATCH] Add File

---
 .../dromara/easyai/resnet/ResnetInput.java    | 40 +++++++++++++++++++
 1 file changed, 40 insertions(+)
 create mode 100644 src/main/java/org/dromara/easyai/resnet/ResnetInput.java

diff --git a/src/main/java/org/dromara/easyai/resnet/ResnetInput.java b/src/main/java/org/dromara/easyai/resnet/ResnetInput.java
new file mode 100644
index 0000000..da8bd35
--- /dev/null
+++ b/src/main/java/org/dromara/easyai/resnet/ResnetInput.java
@@ -0,0 +1,40 @@
+package org.dromara.easyai.resnet;
+
+import org.dromara.easyai.entity.ThreeChannelMatrix;
+import org.dromara.easyai.i.OutBack;
+import org.dromara.easyai.matrixTools.Matrix;
+
+import java.util.ArrayList;
+import java.util.List;
+import java.util.Map;
+
+/**
+ * @author lidapeng
+ * @time 2025/4/17 13:04
+ * @des restNet输出类
+ */
+public class ResnetInput {
+    private final ResBlock fistBlock;//第一个残差块
+    private final int imageSize;//图像大小
+
+    public ResnetInput(ResBlock fistBlock, int imageSize) {
+        this.fistBlock = fistBlock;
+        this.imageSize = imageSize;
+    }
+
+    public void postThreeChannelMatrix(ThreeChannelMatrix threeChannelMatrix, boolean study, Map<Integer, Float> E, OutBack outBack
+            , long eventID, boolean outFeature) throws Exception {
+        int x = threeChannelMatrix.getX();
+        int y = threeChannelMatrix.getY();
+        if (x == y && x == imageSize) {
+            List<Matrix> matrixList = new ArrayList<>();
+            matrixList.add(threeChannelMatrix.getMatrixR());
+            matrixList.add(threeChannelMatrix.getMatrixG());
+            matrixList.add(threeChannelMatrix.getMatrixB());
+            fistBlock.sendMatrixList(matrixList, outBack, study, E, eventID, outFeature);
+        } else {
+            throw new Exception("输入的图像必须为正方形，且尺寸必须等同于初始化配置的值：" + imageSize);
+        }
+    }
+
+}