From 4fa7996c9d78fb80dfd6641659c15af723f3b3ee Mon Sep 17 00:00:00 2001
From: wangbing <wangbing@wbsite.xyz>
Date: Thu, 20 Feb 2025 23:20:57 +0800
Subject: [PATCH] =?UTF-8?q?=E4=B8=8A=E4=BC=A0?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pom.xml                                   |  6 ++++++
 src/main/java/xyz/wbsite/easyai/Main.java | 20 ++++++++++++++++++--
 2 files changed, 24 insertions(+), 2 deletions(-)

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