48 lines
2.4 KiB
YAML
48 lines
2.4 KiB
YAML
|
# global system:
|
||
|
|
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_local_class: 1 # number of classes per client
|
||
|
|
|
||
|
|
res_root: "results" # root directory for results
|
||
|
|
dataset_path: "/home/image1325/ssd1/dataset/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
|
||
|
|
|
||
|
|
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
|
||
|
|
partition_mode: "overlap" # "overlap" or "disjoint"
|
||
|
|
connection_ratio: 1 # connection ratio, e.g., 1.0 means all clients
|
||
|
|
|
||
|
|
# local training:
|
||
|
|
min_lr: 0.000100000000 # initial learning rate
|
||
|
|
max_lr: 0.010000000000 # maximum learning rate
|
||
|
|
momentum: 0.9370000000 # SGD momentum/Adam beta1
|
||
|
|
weight_decay: 0.000500 # optimizer weight decay
|
||
|
|
|
||
|
|
warmup_epochs: 3.00000 # warmup epochs
|
||
|
|
box: 7.500000000000000 # box loss gain
|
||
|
|
cls: 0.500000000000000 # cls loss gain
|
||
|
|
dfl: 1.500000000000000 # dfl loss gain
|
||
|
|
hsv_h: 0.0150000000000 # image HSV-Hue augmentation (fraction)
|
||
|
|
hsv_s: 0.7000000000000 # image HSV-Saturation augmentation (fraction)
|
||
|
|
hsv_v: 0.4000000000000 # image HSV-Value augmentation (fraction)
|
||
|
|
degrees: 0.00000000000 # image rotation (+/- deg)
|
||
|
|
translate: 0.100000000 # image translation (+/- fraction)
|
||
|
|
scale: 0.5000000000000 # image scale (+/- gain)
|
||
|
|
shear: 0.0000000000000 # image shear (+/- deg)
|
||
|
|
flip_ud: 0.00000000000 # image flip up-down (probability)
|
||
|
|
flip_lr: 0.50000000000 # image flip left-right (probability)
|
||
|
|
mosaic: 1.000000000000 # image mosaic (probability)
|
||
|
|
mix_up: 0.000000000000 # image mix-up (probability)
|
||
|
|
names:
|
||
|
|
0: uav
|