dongdonghy / repulsion_loss_pytorch

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

repulsion loss with pytorch in object detection

Overviews

• This is a demo for repulsion loss. You can add it at the final loss calculation step in your own repo.

Dependencies

• python3.6
• numpy
• torch

Inputs

• rois
• boxes_deltas
• groundtruth boxes
• rois_inside_ws
• rois_outside_ws