Add support for DetectoRS. (#3064)
* Add support for DetectoRS.
* Fix Travis CI build failure and remove loading rfp_pretrained for test
* Revise the code according to the PR reviews:
* remove unnecessary `,`
* add docstring to ASPP and ConvAWS2d
* make args of ASPP configurable
* simplify the code of ConvAWS2d
* import nn and functional for SAConv2d
* rewrite rfp_conv with build_conv_layer and constant_init
* rename rfp_inp with rfp_inplanes
* reformat code to pass Travis CI build
* Revise codes according to the PR reviews:
* Create new backbone DetectoRS_ResNe(X)t to incorporate RFP and SAC.
* Initialize modules with mmcv.cnn.
* Revise the codes of RFP.
* Revise codes according to the PR comments:
* Add docstrings to Bottleneck in mmdet/models/backbones/detectors_resnet.py.
* Add docstrings to _load_from_state_dict in mmdet/ops/conv_ws.py.
* Initialize the switch of SAC with mmcv.
* minior refactor and docstring
* update doc
* update README
* move init into init_weights()
* update model zoo
* add gfl
Co-authored-by:
Jiarui XU <xvjiarui0826@gmail.com>
Showing
- README.md 2 additions, 0 deletionsREADME.md
- configs/detectors/README.md 37 additions, 0 deletionsconfigs/detectors/README.md
- configs/detectors/cascade_rcnn_r50_rfp_1x_coco.py 28 additions, 0 deletionsconfigs/detectors/cascade_rcnn_r50_rfp_1x_coco.py
- configs/detectors/cascade_rcnn_r50_sac_1x_coco.py 12 additions, 0 deletionsconfigs/detectors/cascade_rcnn_r50_sac_1x_coco.py
- configs/detectors/detectors_cascade_rcnn_r50_1x_coco.py 32 additions, 0 deletionsconfigs/detectors/detectors_cascade_rcnn_r50_1x_coco.py
- configs/detectors/detectors_htc_r50_1x_coco.py 28 additions, 0 deletionsconfigs/detectors/detectors_htc_r50_1x_coco.py
- configs/detectors/htc_r50_rfp_1x_coco.py 24 additions, 0 deletionsconfigs/detectors/htc_r50_rfp_1x_coco.py
- configs/detectors/htc_r50_sac_1x_coco.py 8 additions, 0 deletionsconfigs/detectors/htc_r50_sac_1x_coco.py
- docs/model_zoo.md 6 additions, 0 deletionsdocs/model_zoo.md
- mmdet/models/backbones/__init__.py 3 additions, 1 deletionmmdet/models/backbones/__init__.py
- mmdet/models/backbones/detectors_resnet.py 305 additions, 0 deletionsmmdet/models/backbones/detectors_resnet.py
- mmdet/models/backbones/detectors_resnext.py 121 additions, 0 deletionsmmdet/models/backbones/detectors_resnext.py
- mmdet/models/necks/__init__.py 3 additions, 1 deletionmmdet/models/necks/__init__.py
- mmdet/models/necks/rfp.py 122 additions, 0 deletionsmmdet/models/necks/rfp.py
- mmdet/ops/__init__.py 3 additions, 1 deletionmmdet/ops/__init__.py
- mmdet/ops/conv_ws.py 98 additions, 0 deletionsmmdet/ops/conv_ws.py
- mmdet/ops/saconv.py 126 additions, 0 deletionsmmdet/ops/saconv.py
- tools/test.py 4 additions, 0 deletionstools/test.py
Loading
Please register or sign in to comment