From fdb70869f97369149505ce5c25a49e3d0b57e0b0 Mon Sep 17 00:00:00 2001 From: Yunhao Meng Date: Fri, 31 Oct 2025 13:14:47 +0800 Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=E9=85=8D=E7=BD=AE=E6=96=87?= =?UTF-8?q?=E4=BB=B6=EF=BC=8C=E7=A7=BB=E9=99=A4=20YOLO=20=E7=89=88?= =?UTF-8?q?=E6=9C=AC=E5=AE=9E=E7=8E=B0=E8=AE=A1=E5=88=92=EF=BC=8C=E8=B0=83?= =?UTF-8?q?=E6=95=B4=20UAV=20=E5=92=8C=20COCO=20=E9=85=8D=E7=BD=AE?= =?UTF-8?q?=E4=B8=AD=E7=9A=84=E6=95=B0=E6=8D=AE=E9=87=8F=E5=92=8C=E6=89=B9?= =?UTF-8?q?=E6=AC=A1=E5=A4=A7=E5=B0=8F?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- README.md | 3 --- config/coco_cfg.yaml | 4 ++-- config/uav_cfg.yaml | 16 ++++++++-------- 3 files changed, 10 insertions(+), 13 deletions(-) diff --git a/README.md b/README.md index d543644..bc87619 100644 --- a/README.md +++ b/README.md @@ -19,9 +19,6 @@ nohup bash fed_run.sh 1 > train.log 2>&1 & - Implement FedProx - Implement SCAFFOLD - Implement FedNova -- Add more YOLO versions (e.g., YOLOv8, YOLOv5, etc.) - - Implement YOLOv8 - - Implement YOLOv5 # references [PyTorch Federated Learning](https://github.com/rruisong/pytorch_federated_learning) diff --git a/config/coco_cfg.yaml b/config/coco_cfg.yaml index 6faa286..0141a84 100644 --- a/config/coco_cfg.yaml +++ b/config/coco_cfg.yaml @@ -17,8 +17,8 @@ local_batch_size: 32 # local training batch size val_batch_size: 128 # validation batch size num_workers: 8 # number of data loader workers -min_data: 1700 # minimum number of images per client -max_data: 1800 # maximum number of images per client +min_data: 1800 # minimum number of images per client +max_data: 1900 # maximum number of images per client partition_mode: "overlap" # "overlap" or "disjoint" connection_ratio: 1 # connection ratio, e.g., 1.0 means all clients diff --git a/config/uav_cfg.yaml b/config/uav_cfg.yaml index 74e8f77..119f3bd 100644 --- a/config/uav_cfg.yaml +++ b/config/uav_cfg.yaml @@ -3,22 +3,22 @@ fed_algo: "FedAvg" # federated learning algorithm model_name: "yolo_v11_n" # yolo_v11_n, yolo_v11_t, yolo_v11_s, yolo_v11_m, yolo_v11_l, yolo_v11_x i_seed: 202509 # initial random seed -num_client: 100 # total number of clients -num_round: 500 # total number of communication rounds +num_client: 36 # total number of clients +num_round: 50 # total number of communication rounds num_local_class: 1 # number of classes per client res_root: "results" # root directory for results -dataset_path: "/home/image1325/ssd1/dataset/uav/" +dataset_path: "/mnt/DATA/uav/" # train_txt: "train.txt" # path to training set txt file # val_txt: "val.txt" # path to validation set txt file # test_txt: "test.txt" # path to test set txt file -local_batch_size: 32 # local training batch size -val_batch_size: 16 # validation batch size +local_batch_size: 36 # local training batch size +val_batch_size: 128 # validation batch size -num_workers: 4 # number of data loader workers -min_data: 640 # minimum number of images per client -max_data: 720 # maximum number of images per client +num_workers: 8 # number of data loader workers +min_data: 385 # minimum number of images per client +max_data: 400 # maximum number of images per client partition_mode: "overlap" # "overlap" or "disjoint" connection_ratio: 1 # connection ratio, e.g., 1.0 means all clients