Encapsulate the second stage into RoI heads (#1999)
* create roi_head * create roi_head mv shared_head into roi_head * fix conflict * refactor(cascade rcnn): create cascade head in roi head, simplify cascade rcnn * type: fix scope: cascade_head subject: mv stage_loss_weight from train_cfg to roi_head creation * type: fix scope: roi_heads subject: handle the case when train_cfg is None * type: fix scope: roi_heads mask inference subject: fix test_cfg.rcnn bug to test_cfg * fix mask eval bug * refactor(double_head): refactor double head roi head * fix(double_head): clean code and fix __init__.py * refactor(mask scoring): create mask scoring roi head * refactor(htc_head): add htc_head * fix(htc_head): add htc_head in __init__.py * fix (htc_head): fix htc_head bugs * fix (ms_roi_head): rm test_cfg.rcnn to test_cfg * feature (grid rcnn): grid roi head * fix (grid_head): fix grid head bug * reformat and change all cfgs * reformat (roi_head): reformat __init__.py for isort=4.3.21 * fix roi_head test bug * fix (carafe): change carafe cfgs to use roi_head * fix (roi_head): fix missing roi_head refactor * reformat to pass CI * test all cfgs * match keys of configs with previous commit 77d073a * add and pass unittest for all roi heads * Refactor (roi_head): extract abstract base class for roi head * Refactor (roi_head): refactor init functions * Refactor (roi_head): weight init * Refactor (roi_head): add _bbox_forwrd & _mask_forward as basic functions * Fix (grid_roi_head): fix bug in bbox_forward_train * Refactor (roi_head): change to use img.device in forward_dummy) * Refactor (roi_heads): simplify init functions and _mask_forward * Fix (cascade_roi_head): fix init bug of cascade roi_head * Refactor (roi_head): use dict as outputs of _bbox_forward and _mask_forward * Refactor (test_config): scan valid configs rather than list them all
Showing
- .gitignore 1 addition, 0 deletions.gitignore
- configs/_base_/models/cascade_mask_rcnn_r50_fpn.py 67 additions, 57 deletionsconfigs/_base_/models/cascade_mask_rcnn_r50_fpn.py
- configs/_base_/models/cascade_rcnn_r50_fpn.py 56 additions, 46 deletionsconfigs/_base_/models/cascade_rcnn_r50_fpn.py
- configs/_base_/models/fast_rcnn_r50_fpn.py 19 additions, 17 deletionsconfigs/_base_/models/fast_rcnn_r50_fpn.py
- configs/_base_/models/faster_rcnn_r50_caffe_c4.py 28 additions, 26 deletionsconfigs/_base_/models/faster_rcnn_r50_caffe_c4.py
- configs/_base_/models/faster_rcnn_r50_fpn.py 19 additions, 17 deletionsconfigs/_base_/models/faster_rcnn_r50_fpn.py
- configs/_base_/models/mask_rcnn_r50_caffe_c4.py 37 additions, 35 deletionsconfigs/_base_/models/mask_rcnn_r50_caffe_c4.py
- configs/_base_/models/mask_rcnn_r50_fpn.py 32 additions, 30 deletionsconfigs/_base_/models/mask_rcnn_r50_fpn.py
- configs/carafe/mask_rcnn_r50_fpn_carafe_1x_coco.py 10 additions, 9 deletionsconfigs/carafe/mask_rcnn_r50_fpn_carafe_1x_coco.py
- configs/cascade_rcnn/cascade_rcnn_x101_64x4d_fpn_1x_coco.py 0 additions, 1 deletionconfigs/cascade_rcnn/cascade_rcnn_x101_64x4d_fpn_1x_coco.py
- configs/cityscapes/faster_rcnn_r50_fpn_1x_cityscapes.py 13 additions, 12 deletionsconfigs/cityscapes/faster_rcnn_r50_fpn_1x_cityscapes.py
- configs/cityscapes/mask_rcnn_r50_fpn_1x_cityscapes.py 21 additions, 20 deletionsconfigs/cityscapes/mask_rcnn_r50_fpn_1x_cityscapes.py
- configs/dcn/faster_rcnn_r50_fpn_dpool_1x_coco.py 12 additions, 11 deletionsconfigs/dcn/faster_rcnn_r50_fpn_dpool_1x_coco.py
- configs/dcn/faster_rcnn_r50_fpn_mdpool_1x_coco.py 12 additions, 11 deletionsconfigs/dcn/faster_rcnn_r50_fpn_mdpool_1x_coco.py
- configs/double_heads/dh_faster_rcnn_r50_fpn_1x_coco.py 19 additions, 18 deletionsconfigs/double_heads/dh_faster_rcnn_r50_fpn_1x_coco.py
- configs/faster_rcnn/faster_rcnn_r50_caffe_fpn_1x_coco.py 2 additions, 1 deletionconfigs/faster_rcnn/faster_rcnn_r50_caffe_fpn_1x_coco.py
- configs/gn+ws/faster_rcnn_r50_fpn_gn_ws-all_1x_coco.py 6 additions, 5 deletionsconfigs/gn+ws/faster_rcnn_r50_fpn_gn_ws-all_1x_coco.py
- configs/gn+ws/mask_rcnn_r50_fpn_gn_ws-all_2x_coco.py 7 additions, 6 deletionsconfigs/gn+ws/mask_rcnn_r50_fpn_gn_ws-all_2x_coco.py
- configs/gn/mask_rcnn_r50_fpn_gn-all_2x_coco.py 6 additions, 5 deletionsconfigs/gn/mask_rcnn_r50_fpn_gn-all_2x_coco.py
- configs/gn/mask_rcnn_r50_fpn_gn-all_contrib_2x_coco.py 6 additions, 5 deletionsconfigs/gn/mask_rcnn_r50_fpn_gn-all_contrib_2x_coco.py
Loading
Please register or sign in to comment