- This is a demo for repulsion loss. You can add it at the final loss calculation step in your own repo.
- python3.6
- numpy
- torch
- rois
- boxes_deltas
- groundtruth boxes
- rois_inside_ws
- rois_outside_ws
repulsion loss with PyTorch:https://arxiv.org/abs/1711.07752
repulsion loss with PyTorch:https://arxiv.org/abs/1711.07752