roi_extractors: GRoIE addition (#2584)
* roi_extractors: GRoIE addition Signed-off-by:Leonardo Rossi <leonardo.rossi@unipr.it> * Fix bug caused by empty tensor and update benchmark results Co-authored-by:
ZwwWayne <wayne.zw@outlook.com>
Showing
- .gitignore 1 addition, 1 deletion.gitignore
- README.md 1 addition, 0 deletionsREADME.md
- configs/grid_rcnn/grid_rcnn_r50_fpn_gn-head_1x_coco.py 11 additions, 0 deletionsconfigs/grid_rcnn/grid_rcnn_r50_fpn_gn-head_1x_coco.py
- configs/groie/README.md 65 additions, 0 deletionsconfigs/groie/README.md
- configs/groie/faster_rcnn_r50_fpn_groie_1x_coco.py 24 additions, 0 deletionsconfigs/groie/faster_rcnn_r50_fpn_groie_1x_coco.py
- configs/groie/grid_rcnn_r50_fpn_gn-head_groie_1x_coco.py 44 additions, 0 deletionsconfigs/groie/grid_rcnn_r50_fpn_gn-head_groie_1x_coco.py
- configs/groie/mask_rcnn_r101_fpn_syncbn-backbone_r4_gcb_c3-c5_groie_1x_coco.py 44 additions, 0 deletions...nn_r101_fpn_syncbn-backbone_r4_gcb_c3-c5_groie_1x_coco.py
- configs/groie/mask_rcnn_r50_fpn_groie_1x_coco.py 44 additions, 0 deletionsconfigs/groie/mask_rcnn_r50_fpn_groie_1x_coco.py
- configs/groie/mask_rcnn_r50_fpn_syncbn-backbone_r4_gcb_c3-c5_groie_1x_coco.py 44 additions, 0 deletions...cnn_r50_fpn_syncbn-backbone_r4_gcb_c3-c5_groie_1x_coco.py
- docs/model_zoo.md 3 additions, 0 deletionsdocs/model_zoo.md
- mmdet/models/roi_heads/roi_extractors/__init__.py 5 additions, 1 deletionmmdet/models/roi_heads/roi_extractors/__init__.py
- mmdet/models/roi_heads/roi_extractors/groie.py 62 additions, 0 deletionsmmdet/models/roi_heads/roi_extractors/groie.py
- mmdet/ops/plugin.py 4 additions, 1 deletionmmdet/ops/plugin.py
- tests/test_roi_extractor.py 41 additions, 0 deletionstests/test_roi_extractor.py
Loading
Please register or sign in to comment