Files
fed-yolo/README.md

591 B

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