- 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
-
Notifications
You must be signed in to change notification settings - Fork 9
repulsion loss with PyTorch：https://arxiv.org/abs/1711.07752
dongdonghy/repulsion_loss_pytorch
Folders and files
| Name | Name | Last commit message | Last commit date | |
|---|---|---|---|---|
Repository files navigation
Releases
No releases published
Packages 0
No packages published