diff --git a/src/main/java/org/dromara/easyai/unet/UNetModel.java b/src/main/java/org/dromara/easyai/unet/UNetModel.java
new file mode 100644
index 0000000..87b47c5
--- /dev/null
+++ b/src/main/java/org/dromara/easyai/unet/UNetModel.java
@@ -0,0 +1,29 @@
+package org.dromara.easyai.unet;
+
+import java.util.List;
+
+/**
+ * @author lidapeng
+ * @time 2025/3/3 15:00
+ * @des UNet模型
+ */
+public class UNetModel {
+    private List<ConvModel> encoderModels;
+    private List<ConvModel> decoderModels;
+
+    public List<ConvModel> getEncoderModels() {
+        return encoderModels;
+    }
+
+    public void setEncoderModels(List<ConvModel> encoderModels) {
+        this.encoderModels = encoderModels;
+    }
+
+    public List<ConvModel> getDecoderModels() {
+        return decoderModels;
+    }
+
+    public void setDecoderModels(List<ConvModel> decoderModels) {
+        this.decoderModels = decoderModels;
+    }
+}