Code for "Grid R-CNN" (#810)
* Grid R-CNN * add grid_rcnn_res50fpn2x config * add assertion that grid_head should exist * fix bugs and remove SharedFCBBoxHeadGrid * remove the property with_grid * format fixes for grad_head and add config dir * move random_jitter to grid_head and some refactoring * simplify the calculation of num_edges * refactoring * refactoring * rename config files and add x101 config * bug fix for inference * remove random_jitter_single * add readme of grid rcnn * add bibtex of grid rcnn plus * update work_dir
Showing
- configs/grid_rcnn/README.md 33 additions, 0 deletionsconfigs/grid_rcnn/README.md
- configs/grid_rcnn/grid_rcnn_gn_head_r50_fpn_2x.py 178 additions, 0 deletionsconfigs/grid_rcnn/grid_rcnn_gn_head_r50_fpn_2x.py
- configs/grid_rcnn/grid_rcnn_gn_head_x101_32x4d_fpn_2x.py 180 additions, 0 deletionsconfigs/grid_rcnn/grid_rcnn_gn_head_x101_32x4d_fpn_2x.py
- mmdet/models/bbox_heads/bbox_head.py 4 additions, 2 deletionsmmdet/models/bbox_heads/bbox_head.py
- mmdet/models/detectors/__init__.py 2 additions, 1 deletionmmdet/models/detectors/__init__.py
- mmdet/models/detectors/grid_rcnn.py 205 additions, 0 deletionsmmdet/models/detectors/grid_rcnn.py
- mmdet/models/mask_heads/__init__.py 5 additions, 1 deletionmmdet/models/mask_heads/__init__.py
- mmdet/models/mask_heads/grid_head.py 359 additions, 0 deletionsmmdet/models/mask_heads/grid_head.py
Loading
Please register or sign in to comment