From 2a3e5b17e795d03adcf8457cee1d58d3915506c9 Mon Sep 17 00:00:00 2001
From: Yunhao Meng <yunhao.meng@outlook.com>
Date: Mon, 5 May 2025 17:30:12 +0800
Subject: [PATCH] =?UTF-8?q?yolov8=E5=AF=B9=E6=AF=94=E8=AE=AD=E7=BB=83?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 yolov8/yolov8.yaml     |  6 ++++++
 yolov8/yolov8_train.py | 13 +++++++++++++
 2 files changed, 19 insertions(+)
 create mode 100644 yolov8/yolov8.yaml
 create mode 100644 yolov8/yolov8_train.py

diff --git a/yolov8/yolov8.yaml b/yolov8/yolov8.yaml
new file mode 100644
index 0000000..e8d137f
--- /dev/null
+++ b/yolov8/yolov8.yaml
@@ -0,0 +1,6 @@
+train: /mnt/DATA/dataset/uav_dataset/train/images/
+val:   /mnt/DATA/dataset/uav_dataset/val/images/
+test:  /mnt/DATA/dataset/test2/images/
+# number of classes
+nc:    1
+names: ['uav']
\ No newline at end of file
diff --git a/yolov8/yolov8_train.py b/yolov8/yolov8_train.py
new file mode 100644
index 0000000..de0f4e1
--- /dev/null
+++ b/yolov8/yolov8_train.py
@@ -0,0 +1,13 @@
+from ultralytics import YOLO
+
+# 加载预训练模型
+model = YOLO('../yolov8n.pt')
+
+# 开始训练
+model.train(
+    data='./yolov8.yaml',  # 数据配置文件路径
+    epochs=320,                 # 训练轮数
+    batch=-1,                  # 批量大小
+    imgsz=640,                 # 输入图片大小
+    device=0                   # 使用的设备（0 表示 GPU，'cpu' 表示 CPU）
+)
\ No newline at end of file