Code for "Double-Head RCNN: Rethinking Classification and Localization for Object Detection" (#809)
* add major components for training double head double head only with two losses and no attention remove double_head detector for now, merge upchannel to double_head bbox head change the stype using yapl remove uncessary comment# to pass check to pass v2 to pass v3 line too long and style again reuse bottlenet * refactoring * bug fix * bug fix for weight initialization * add reg roi scale factor and modify loss weights * rescale the roi after mapping to fpn levels
Showing
- configs/double_heads/dh_faster_rcnn_r50_fpn_1x.py 170 additions, 0 deletionsconfigs/double_heads/dh_faster_rcnn_r50_fpn_1x.py
- mmdet/models/bbox_heads/__init__.py 4 additions, 1 deletionmmdet/models/bbox_heads/__init__.py
- mmdet/models/bbox_heads/double_bbox_head.py 170 additions, 0 deletionsmmdet/models/bbox_heads/double_bbox_head.py
- mmdet/models/detectors/__init__.py 2 additions, 1 deletionmmdet/models/detectors/__init__.py
- mmdet/models/detectors/double_head_rcnn.py 154 additions, 0 deletionsmmdet/models/detectors/double_head_rcnn.py
- mmdet/models/roi_extractors/single_level.py 17 additions, 1 deletionmmdet/models/roi_extractors/single_level.py
Loading
Please register or sign in to comment