-
- Downloads
Reimplement "Mask Scoring R-CNN" (#621)
* add ms-rcnn * fix config * fix bug in test * fix bug in test, det_labels should plus 1 * add area ratio * speed up area ratio calculation * delete caffe style config * refactor code, bbox and segm eval using different score * fix bug in distevalhook * update README and config * update x101-64x4d * refactor code * update upstream * format * update r50 and r101 2x performance * update ms-rcnn-x101-64x4d-2x performance * move simple_mask_test, add docstring and comment * delete TODO in mask iou target * add detail comment to mask iou target * modify comment of mask iou target * move mask iou target to mask iou head * merge mask_iou_target and get_target * add MSELoss for MaskIoUHead and rename mask_iou to fc_mask_iou * refactoring * fix bug in test * fix bug when cropping mask inside proposal * update readme * update bibtex * remove space in bibtex
Showing
- configs/ms_rcnn/README.md 23 additions, 0 deletionsconfigs/ms_rcnn/README.md
- configs/ms_rcnn/ms_rcnn_r101_caffe_fpn_1x.py 192 additions, 0 deletionsconfigs/ms_rcnn/ms_rcnn_r101_caffe_fpn_1x.py
- configs/ms_rcnn/ms_rcnn_r50_caffe_fpn_1x.py 192 additions, 0 deletionsconfigs/ms_rcnn/ms_rcnn_r50_caffe_fpn_1x.py
- configs/ms_rcnn/ms_rcnn_x101_64x4d_fpn_1x.py 193 additions, 0 deletionsconfigs/ms_rcnn/ms_rcnn_x101_64x4d_fpn_1x.py
- mmdet/models/detectors/__init__.py 2 additions, 1 deletionmmdet/models/detectors/__init__.py
- mmdet/models/detectors/mask_scoring_rcnn.py 197 additions, 0 deletionsmmdet/models/detectors/mask_scoring_rcnn.py
- mmdet/models/losses/__init__.py 4 additions, 3 deletionsmmdet/models/losses/__init__.py
- mmdet/models/losses/mse_loss.py 25 additions, 0 deletionsmmdet/models/losses/mse_loss.py
- mmdet/models/mask_heads/__init__.py 3 additions, 2 deletionsmmdet/models/mask_heads/__init__.py
- mmdet/models/mask_heads/maskiou_head.py 181 additions, 0 deletionsmmdet/models/mask_heads/maskiou_head.py
Loading
Please register or sign in to comment