/repulsion_loss_pytorch

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

Primary LanguagePython

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