diff --git a/src/main/java/org/dromara/easyai/yolo/YoloModel.java b/src/main/java/org/dromara/easyai/yolo/YoloModel.java
new file mode 100644
index 0000000..1c8a60f
--- /dev/null
+++ b/src/main/java/org/dromara/easyai/yolo/YoloModel.java
@@ -0,0 +1,28 @@
+package org.dromara.easyai.yolo;
+
+import org.dromara.easyai.nerveCenter.ModelParameter;
+import org.dromara.easyai.resnet.entity.ResnetModel;
+
+import java.util.List;
+
+public class YoloModel {
+    private List<TypeModel> typeModels;
+    private ModelParameter typeModel;
+
+    public List<TypeModel> getTypeModels() {
+        return typeModels;
+    }
+
+    public void setTypeModels(List<TypeModel> typeModels) {
+        this.typeModels = typeModels;
+    }
+
+    public ModelParameter getTypeModel() {
+        return typeModel;
+    }
+
+    public void setTypeModel(ModelParameter typeModel) {
+        this.typeModel = typeModel;
+    }
+
+}