where to add repulsion loss

dongdonghy / repulsion_loss_pytorch

repulsion loss with PyTorch：https://arxiv.org/abs/1711.07752

16 stars 9 forks source link

where to add repulsion loss #1

Open buptzqr opened 5 years ago

buptzqr commented 5 years ago

hello，I want to ask we should add repulsion loss to final loss only or we should add repulsion loss to rpn loss and final loss both?

XingLiuJia commented 5 years ago

hello，I want to ask we should add repulsion loss to final loss only or we should add repulsion loss to rpn loss and final loss both?

你好，你是在问如何加在two-stage上吗？

ww451575464 commented 5 years ago

我想加在faster rcnn中，这个loss是不是值替换最后的rcnn bbox loss的那个smooth l1 loss就可以了，rpn bbox loss还是保持原样，此外，repulsion loss似乎应该是三个loss之和，第一个attr loss没有，是否就是原本的smooth l1 loss呢