Code of "GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond" (#598)
* add pytorch 1.1.0 SyncBN support * change BatchNorm2d to _BatchNorm and call freeze after train * add freeze back to init function * fixed indentation typo in adding freeze * convert SyncBN to BN when eval is called * add gcb and configs * fixed line too long * fixed space and newline * ignore too long line in config files * two space before inline comment * refactor var name * fixed line too long * changed SyncBN to backbone only and add non-Sync BN configs * reformat config to adapt master branch * reformat change method->type * remoe some configs * clean up readme.md * add benchmark * fix typo in README * change sbn to syncbn * change work dir * refactor gcb * fixed line too long * Code formatting
Showing
- configs/gcnet/README.md 58 additions, 0 deletionsconfigs/gcnet/README.md
- configs/gcnet/mask_rcnn_r16_gcb_c3-c5_r50_fpn_1x.py 185 additions, 0 deletionsconfigs/gcnet/mask_rcnn_r16_gcb_c3-c5_r50_fpn_1x.py
- configs/gcnet/mask_rcnn_r16_gcb_c3-c5_r50_fpn_syncbn_1x.py 189 additions, 0 deletionsconfigs/gcnet/mask_rcnn_r16_gcb_c3-c5_r50_fpn_syncbn_1x.py
- configs/gcnet/mask_rcnn_r4_gcb_c3-c5_r50_fpn_1x.py 185 additions, 0 deletionsconfigs/gcnet/mask_rcnn_r4_gcb_c3-c5_r50_fpn_1x.py
- configs/gcnet/mask_rcnn_r4_gcb_c3-c5_r50_fpn_syncbn_1x.py 189 additions, 0 deletionsconfigs/gcnet/mask_rcnn_r4_gcb_c3-c5_r50_fpn_syncbn_1x.py
- configs/gcnet/mask_rcnn_r50_fpn_sbn_1x.py 185 additions, 0 deletionsconfigs/gcnet/mask_rcnn_r50_fpn_sbn_1x.py
- mmdet/models/backbones/resnet.py 33 additions, 7 deletionsmmdet/models/backbones/resnet.py
- mmdet/models/backbones/resnext.py 9 additions, 4 deletionsmmdet/models/backbones/resnext.py
- mmdet/ops/__init__.py 2 additions, 1 deletionmmdet/ops/__init__.py
- mmdet/ops/gcb/__init__.py 5 additions, 0 deletionsmmdet/ops/gcb/__init__.py
- mmdet/ops/gcb/context_block.py 104 additions, 0 deletionsmmdet/ops/gcb/context_block.py
Loading
Please register or sign in to comment