diff --git a/pom.xml b/pom.xml
index da71e82..0e538c3 100644
--- a/pom.xml
+++ b/pom.xml
@@ -38,5 +38,11 @@
             <artifactId>hutool-all</artifactId>
             <version>5.8.0.M2</version>
         </dependency>
+
+        <dependency>
+            <groupId>org.dromara.easyai</groupId>
+            <artifactId>easyAi</artifactId>
+            <version>1.3.3</version>
+        </dependency>
     </dependencies>
 </project>
\ No newline at end of file
diff --git a/src/main/java/xyz/wbsite/easyai/Main.java b/src/main/java/xyz/wbsite/easyai/Main.java
index 50764a6..bb3cda3 100644
--- a/src/main/java/xyz/wbsite/easyai/Main.java
+++ b/src/main/java/xyz/wbsite/easyai/Main.java
@@ -1,11 +1,27 @@
 package xyz.wbsite.easyai;
 
+import cn.hutool.json.JSONUtil;
+import org.dromara.easyai.yolo.FastYolo;
+import org.dromara.easyai.yolo.YoloConfig;
+import org.dromara.easyai.yolo.YoloModel;
+import org.dromara.easyai.yolo.YoloSample;
+
+import java.util.ArrayList;
+import java.util.List;
+
 /**
  * 主函数入口
  */
 public class Main {
 
-    public static void main(String[] args) {
-        System.out.println("Hello world!");
+    public static void main(String[] args) throws Exception {
+        // 训练
+        YoloConfig yoloConfig = new YoloConfig();
+        yoloConfig.setShowLog(true);
+        List<YoloSample> yoloSamples = new ArrayList<>();
+        FastYolo fastYolo = new FastYolo(yoloConfig);
+        fastYolo.toStudy(yoloSamples);
+        YoloModel yoloModel = fastYolo.getModel();
+        System.out.println(JSONUtil.toJsonStr(yoloModel));
     }
 }
\ No newline at end of file