.gitignore

@@ -299,6 +299,5 @@ Temporary Items
 
 # project files
 /whl_packages/
-runs/
-*.pt
+/federated_learning/runs/detect/*
 *.cache

yolov8/yolov8.yaml

@@ -1,6 +0,0 @@
-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']

yolov8/yolov8_train.py

@@ -1,13 +0,0 @@
-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）
-)