30 lines
591 B
Markdown
30 lines
591 B
Markdown
# fed-yolo
|
|
|
|
Combine Federated Learning with YOLOv11.
|
|
|
|
## requirements
|
|
```bash
|
|
pip install -r requirements.txt
|
|
```
|
|
|
|
## how to run
|
|
```bash
|
|
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](https://github.com/rruisong/pytorch_federated_learning)
|
|
|
|
|
|
[YOLOv11-pt](https://github.com/jahongir7174/YOLOv11-pt) |