WZMIAOMIAO / deep-learning-for-image-processing

deep learning for image processing including classification and object-detection etc.
GNU General Public License v3.0
22.41k stars 7.91k forks source link

iou #739

Open thestars-maker opened 1 year ago

thestars-maker commented 1 year ago

有没有大佬知道怎么改IOU损失呢,如果我不想使用标准的IOU计算想采用ciou,siou等,除了修改box.py中的box_iou函数。这个函数返回的是CIOU,SIOU。还需要对那些地方修改哦

HaronCHou commented 1 year ago

yolov3-spp中有giou diou ciou的计算代码