Fedavg and YOLOv11 training

config/uav_cfg.yaml

@@ -0,0 +1,47 @@
+# 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