search

Zzh-tju / CIoU

Distance-IoU Loss: Faster and Better Learning for Bounding Box Regression (AAAI 2020)
GNU General Public License v3.0
319 stars 44 forks source link

No `distance` function #5

Closed thuyngch closed 4 years ago

thuyngch commented 4 years ago

Hi @Zzh-tju , Thanks for your great work of CIoU and Cluster NMS.

I noted that the Cluster NMS is so impressed. Without any modification in the model, by changing the last NMS, the mAP is improved significantly. However, at https://github.com/Zzh-tju/CIoU/blob/master/layers/functions/detection.py#L3, the file https://github.com/Zzh-tju/CIoU/blob/master/layers/box_utils.py does not have distance function. Could you help me to figure out that?

Thanks,

Zzh-tju commented 4 years ago

updated. closed.