polygon_inter_union_cuda result error.

XinzeLee / PolygonObjectDetection

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

352 stars 95 forks source link

polygon_inter_union_cuda result error. #4

Open lacoboi opened 3 years ago

lacoboi commented 3 years ago

Thanks for your code ! However there is a bug in polygon_inter_union_cuda function , two boxes which like [ 306.88382, 1386.51160, 1008.65216, 1229.87756, 1242.48157, 2030.92688, 548.18250, 2043.93970 ] and [ 301.47177, 1393.00049, 551.86237, 2047.00061, 1275.99512, 2047.00061, 1007.57446, 1223.00012 ] , in gpu function the iou of them is 0 while the cpu function the iou is 0.95783 , would you please try fix it , thanks.

XinzeLee commented 3 years ago

Thanks for your interests in Polygon-Yolov5. I checked the code, and indeed there exists this problem. And I have modified the codes. Please see 8004b31.