XinzeLee / PolygonObjectDetection

This repository is based on Ultralytics/yolov5, with adjustments to enable polygon prediction boxes.
352 stars 95 forks source link

I encountered a bug during training #24

Open 18112330636 opened 2 years ago

18112330636 commented 2 years ago

Your model uses CIOU during training, but the calculation of poly IOU is fundamentally different from that of rectangle. The CIOU in the code is wrong, but it will affect the convergence. In addition, if I remove the loss of the order of points, it will not affect the convergence at all.