Support PointRend (#2752)
* Add PointRend * remove PointExtractor * add 1x 3x mstrain * fixed test * fixed test bug * add README.md * fixed doc string * update doc * update doc * update comments * fixed docs * update docs * change roi extractor * update docstring * minior update * update docstring * rename to mask point head * simplify super * move point head * rename
Showing
- configs/point_rend/README.md 20 additions, 0 deletionsconfigs/point_rend/README.md
- configs/point_rend/point_rend_r50_caffe_fpn_mstrain_1x_coco.py 41 additions, 0 deletions...gs/point_rend/point_rend_r50_caffe_fpn_mstrain_1x_coco.py
- configs/point_rend/point_rend_r50_caffe_fpn_mstrain_3x_coco.py 4 additions, 0 deletions...gs/point_rend/point_rend_r50_caffe_fpn_mstrain_3x_coco.py
- mmdet/models/detectors/__init__.py 2 additions, 1 deletionmmdet/models/detectors/__init__.py
- mmdet/models/detectors/point_rend.py 29 additions, 0 deletionsmmdet/models/detectors/point_rend.py
- mmdet/models/roi_heads/__init__.py 4 additions, 3 deletionsmmdet/models/roi_heads/__init__.py
- mmdet/models/roi_heads/mask_heads/__init__.py 3 additions, 1 deletionmmdet/models/roi_heads/mask_heads/__init__.py
- mmdet/models/roi_heads/mask_heads/coarse_mask_head.py 91 additions, 0 deletionsmmdet/models/roi_heads/mask_heads/coarse_mask_head.py
- mmdet/models/roi_heads/mask_heads/fcn_mask_head.py 4 additions, 2 deletionsmmdet/models/roi_heads/mask_heads/fcn_mask_head.py
- mmdet/models/roi_heads/mask_heads/mask_point_head.py 300 additions, 0 deletionsmmdet/models/roi_heads/mask_heads/mask_point_head.py
- mmdet/models/roi_heads/mask_scoring_roi_head.py 1 addition, 1 deletionmmdet/models/roi_heads/mask_scoring_roi_head.py
- mmdet/models/roi_heads/point_rend_roi_head.py 194 additions, 0 deletionsmmdet/models/roi_heads/point_rend_roi_head.py
- mmdet/ops/__init__.py 12 additions, 34 deletionsmmdet/ops/__init__.py
- mmdet/ops/point_sample.py 218 additions, 0 deletionsmmdet/ops/point_sample.py
- tests/test_config.py 8 additions, 3 deletionstests/test_config.py
Loading
Please register or sign in to comment