Giters

TL-System / plato

A federated learning framework to support scalable and reproducible research

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

[BUG]

CQIITLAB opened this issue · comments

commented

We tried 500 epochs and the test results are still zero. Then we checked the code and found that NMS was not used on line 310 of "Plato /trainers/yolo.py" and the shape of 'pred' in line 312 was different from the shape of 'pred' in the original Yolov5. After we retrain by adding NMS, the results of 100 and 300 epochs are as follows:
image

image

commented

If you know what the issue is, could you please submit a PR as a fix?

commented

I haven't figured out what the issue is yet. Trying to locate it