Loss function about predicting four coordinates

XinzeLee / PolygonObjectDetection

This repository is based on Ultralytics/yolov5, with adjustments to enable polygon prediction boxes.

352 stars 95 forks source link

Loss function about predicting four coordinates #13

Open bigonestep opened 2 years ago

bigonestep commented 2 years ago

Hi, by looking at your source code, is the anchor box not working in the polygon prediction? Would it be better to change the prediction based on center point offset to the prediction based on anchor frame coordinate offset?