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); + } +}