yolov8对比训练

忽略训练结果和pt文件
2025-05-05 17:30:12 +08:00 · 2025-05-05 17:29:58 +08:00
3 changed files with 21 additions and 1 deletions
--- a/.gitignore
+++ b/.gitignore
@ -299,5 +299,6 @@ Temporary Items

 # project files
 /whl_packages/
-/federated_learning/runs/detect/*
+runs/
+*.pt
 *.cache
--- a/yolov8/yolov8.yaml
+++ 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']
--- a/yolov8/yolov8_train.py
+++ 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）
+)
Author	SHA1	Message	Date
Yunhao Meng	2a3e5b17e7	yolov8对比训练	2025-05-05 17:30:12 +08:00
Yunhao Meng	c57c8f3552	忽略训练结果和pt文件	2025-05-05 17:29:58 +08:00