fed-yolo

Combine Federated Learning with YOLOv11.

requirements

pip install -r requirements.txt

how to run

nohup bash fed_run.sh 1 > train.log 2>&1 &

results

TODO

• Add more FL algorithms (e.g., FedProx, FedAvgM, etc.)
  • Implement FedProx
  • Implement SCAFFOLD
  • Implement FedNova
• Add more YOLO versions (e.g., YOLOv8, YOLOv5, etc.)
  • Implement YOLOv8
  • Implement YOLOv5

references

PyTorch Federated Learning

YOLOv11-pt