Code for AAAI 2021 paper. "SCNet: Training Inference Sample Consistency for...
Code for AAAI 2021 paper. "SCNet: Training Inference Sample Consistency for Instance Segmentation" (#4356) * add scnet * add more configs and minors * fix unittest for scnet * update docstring * add forward test for htc and scnet * fix build on pytorch 1.3 for empty tensor * update doc string and minor refactor * update unit-test and use inheritance for SCNet heads * support scnet tta * minor * update readme * minor fixes
Showing
- README.md 1 addition, 0 deletionsREADME.md
- configs/scnet/README.md 51 additions, 0 deletionsconfigs/scnet/README.md
- configs/scnet/scnet_r101_fpn_20e_coco.py 2 additions, 0 deletionsconfigs/scnet/scnet_r101_fpn_20e_coco.py
- configs/scnet/scnet_r50_fpn_1x_coco.py 136 additions, 0 deletionsconfigs/scnet/scnet_r50_fpn_1x_coco.py
- configs/scnet/scnet_r50_fpn_20e_coco.py 4 additions, 0 deletionsconfigs/scnet/scnet_r50_fpn_20e_coco.py
- configs/scnet/scnet_x101_64x4d_fpn_20e_coco.py 14 additions, 0 deletionsconfigs/scnet/scnet_x101_64x4d_fpn_20e_coco.py
- configs/scnet/scnet_x101_64x4d_fpn_8x1_20e_coco.py 3 additions, 0 deletionsconfigs/scnet/scnet_x101_64x4d_fpn_8x1_20e_coco.py
- mmdet/models/detectors/__init__.py 3 additions, 1 deletionmmdet/models/detectors/__init__.py
- mmdet/models/detectors/scnet.py 10 additions, 0 deletionsmmdet/models/detectors/scnet.py
- mmdet/models/roi_heads/__init__.py 10 additions, 4 deletionsmmdet/models/roi_heads/__init__.py
- mmdet/models/roi_heads/bbox_heads/__init__.py 3 additions, 1 deletionmmdet/models/roi_heads/bbox_heads/__init__.py
- mmdet/models/roi_heads/bbox_heads/scnet_bbox_head.py 76 additions, 0 deletionsmmdet/models/roi_heads/bbox_heads/scnet_bbox_head.py
- mmdet/models/roi_heads/mask_heads/__init__.py 6 additions, 1 deletionmmdet/models/roi_heads/mask_heads/__init__.py
- mmdet/models/roi_heads/mask_heads/feature_relay_head.py 55 additions, 0 deletionsmmdet/models/roi_heads/mask_heads/feature_relay_head.py
- mmdet/models/roi_heads/mask_heads/global_context_head.py 101 additions, 0 deletionsmmdet/models/roi_heads/mask_heads/global_context_head.py
- mmdet/models/roi_heads/mask_heads/scnet_mask_head.py 27 additions, 0 deletionsmmdet/models/roi_heads/mask_heads/scnet_mask_head.py
- mmdet/models/roi_heads/mask_heads/scnet_semantic_head.py 27 additions, 0 deletionsmmdet/models/roi_heads/mask_heads/scnet_semantic_head.py
- mmdet/models/roi_heads/scnet_roi_head.py 582 additions, 0 deletionsmmdet/models/roi_heads/scnet_roi_head.py
- mmdet/models/utils/__init__.py 2 additions, 2 deletionsmmdet/models/utils/__init__.py
- mmdet/models/utils/res_layer.py 85 additions, 0 deletionsmmdet/models/utils/res_layer.py
Loading
Please register or sign in to comment