[BUG]

[CQIITLAB]

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:

[baochunli]

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

[CQIITLAB]

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