From 0b26e29f0ba85e64ab8c757d8d7001386c2797b0 Mon Sep 17 00:00:00 2001 From: inter Date: Thu, 4 Sep 2025 14:09:40 +0800 Subject: [PATCH] Add File --- .../org/dromara/easyai/unet/UNetInput.java | 22 +++++++++++++++++++ 1 file changed, 22 insertions(+) create mode 100644 src/main/java/org/dromara/easyai/unet/UNetInput.java diff --git a/src/main/java/org/dromara/easyai/unet/UNetInput.java b/src/main/java/org/dromara/easyai/unet/UNetInput.java new file mode 100644 index 0000000..9e58be6 --- /dev/null +++ b/src/main/java/org/dromara/easyai/unet/UNetInput.java @@ -0,0 +1,22 @@ +package org.dromara.easyai.unet; + +import org.dromara.easyai.entity.ThreeChannelMatrix; +import org.dromara.easyai.i.OutBack; + +/** + * @author lidapeng + * @time 2025/3/3 16:48 + * @des UNet输入入口类 + */ +public class UNetInput { + private final UNetEncoder uNetEncoder; + + public UNetInput(UNetEncoder uNetEncoder) { + this.uNetEncoder = uNetEncoder; + } + + public void postMessage(long eventID, OutBack outBack, ThreeChannelMatrix feature, ThreeChannelMatrix featureE, + boolean study) throws Exception { + uNetEncoder.sendThreeChannel(eventID, outBack, feature, featureE, study); + } +}