Adopt CIOU/DIOU as a module

[NoOneUST]

Hello,

I am writing a instance segmentation network and I adopt a SSD head as the detection head. If I want to adopt CIOU/DIOU, how should I modify my code to integrate them together? Can this repo be directly called as a module? Thank you!

[Zzh-tju]

I dont know how to integrate the models together. If you want to use D/CIoU loss, you just need to modify the box output to caculate the loss.

[NoOneUST]

Can I just use class MultiBoxLoss(nn.Module) in this file to replace my original detection loss and everything is done? https://github.com/Zzh-tju/DIoU-SSD-pytorch/blob/master/utils/loss/multibox_loss.py

[Zzh-tju]

feel free to try anything until you can get a stable training. my advice may not be helpful