DIoU-pytorch-detectron icon indicating copy to clipboard operation
DIoU-pytorch-detectron copied to clipboard

Published 20 hours ago • verified publisher icon Zzh-tju

i try to implement ciou loss in mmdetection

Open kiddliao opened this issue 5 years ago • 3 comments

shoule i change both roi's loss_bbox and rpn's loss_bbox from l1 loss to ciou loss ,and for doule shared head ?

kiddliao avatar May 24 '20 06:05 kiddliao

CIoU is for bbox regression. Of course you can.

Zzh-tju avatar May 24 '20 06:05 Zzh-tju

谢谢

kiddliao avatar May 24 '20 06:05 kiddliao

shoule i change both roi's loss_bbox and rpn's loss_bbox from l1 loss to ciou loss ,and for doule shared head ?

您好，请问您是怎么使用CIOU的？仅仅是将rpn_head和roi_head里的L1loss替换成了CIOU吗？需要有其他地方的改动吗？我直接修改了这两处的loss，但是训练结果一开始就是nan

gufan-d avatar Mar 15 '21 13:03 gufan-d