Anchor-x/fed-yolo
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

配置更改

Browse Source
This commit is contained in:
Yunhao Meng
2025-10-23 13:07:55 +08:00
parent b81f33ad28
commit c7a3d34d88
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
config/coco_cfg.yaml
View File

@@ -4,21 +4,21 @@ model_name: "yolo_v11_n" # yolo_v11_n, yolo_v11_t, yolo_v11_s, yolo_v11
i_seed: 202509 # initial random seed i_seed: 202509 # initial random seed
num_client: 64 # total number of clients num_client: 64 # total number of clients
num_round: 5 # total number of communication rounds num_round: 50 # total number of communication rounds
num_local_class: 80 # number of classes per client num_local_class: 80 # number of classes per client
res_root: "results" # root directory for results res_root: "results" # root directory for results
dataset_path: "/home/image1325/ssd1/dataset/COCO128/" dataset_path: "/mnt/DATA/coco" # root directory for dataset
# train_txt: "train.txt" # path to training set txt file # train_txt: "train.txt" # path to training set txt file
# val_txt: "val.txt" # path to validation set txt file # val_txt: "val.txt" # path to validation set txt file
# test_txt: "test.txt" # path to test set txt file # test_txt: "test.txt" # path to test set txt file
local_batch_size: 32 # local training batch size local_batch_size: 32 # local training batch size
val_batch_size: 4 # validation batch size val_batch_size: 128 # validation batch size
num_workers: 4 # number of data loader workers num_workers: 8 # number of data loader workers
min_data: 128 # minimum number of images per client min_data: 1700 # minimum number of images per client
max_data: 128 # maximum number of images per client max_data: 1800 # maximum number of images per client
partition_mode: "overlap" # "overlap" or "disjoint" partition_mode: "overlap" # "overlap" or "disjoint"
connection_ratio: 1 # connection ratio, e.g., 1.0 means all clients connection_ratio: 1 # connection ratio, e.g., 1.0 means all clients

4
utils/args.py
View File

@@ -5,9 +5,9 @@ import os
def args_parser(): def args_parser():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--epochs", type=int, default=10, help="number of rounds of local training") parser.add_argument("--epochs", type=int, default=16, help="number of rounds of local training")
parser.add_argument("--input_size", type=int, default=640, help="image input size") parser.add_argument("--input_size", type=int, default=640, help="image input size")
parser.add_argument("--config", type=str, default="./config/uav_cfg.yaml", help="Path to YAML config") parser.add_argument("--config", type=str, default="./config/coco_cfg.yaml", help="Path to YAML config")
args = parser.parse_args() args = parser.parse_args()