Reimplement Sparse R-CNN (#4219)
* draft version train and test * remove get_bbox in sparse_head * refactor SparseRCNN model, align inference eval mAP with official repo * add focal_loss in hungarian_assigner.py * update hungarian_assigner.py * update detr config * update doc and codes * fix unitest * fix unitest * add embedding_rpn and traing code * modify code format * fix proposalist in simpletest * add focal loss traing * fix acc * fix reduce mean * add embedding rpn doc and add show initproposals * add docstr * fix backbone lr * fix typo and change detach_proposal_list to proposal_list * remove debug assert * add readme * add config for random crop when use 300 proposals * add aug for 300 proposals * add 300 proposals results and fix docstr * add 1x results * fix lint * fix unitest * fix docstr of embeddingrpn * fix according to the comments * fix get_bbox and a bug of dynamic conv * support cpu focal loss and fix docstr * fix docstr * fix unitest * fix bug of init_weight * fix to device in assign * fix all commemt * fix test_config * remove dim 1 in forward * fix shape in comment * fix the case that train_cfg is None * fix shape of docstr * fix docstr * fix typo * simplify test * fix docstr * move train_cfg and test_cfg to model as * fix unitest * add link of ckpt and log * fix 1x results Co-authored-by:tianyuandu <tianyuandu@gmail.com> Co-authored-by:
fangkairen <fangkairen@sensetime.com>
Showing
- configs/sparse_rcnn/README.md 28 additions, 0 deletionsconfigs/sparse_rcnn/README.md
- configs/sparse_rcnn/sparse_rcnn_r101_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py 3 additions, 0 deletions...nn_r101_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py
- configs/sparse_rcnn/sparse_rcnn_r101_fpn_mstrain_480-800_3x_coco.py 3 additions, 0 deletions...arse_rcnn/sparse_rcnn_r101_fpn_mstrain_480-800_3x_coco.py
- configs/sparse_rcnn/sparse_rcnn_r50_fpn_1x_coco.py 95 additions, 0 deletionsconfigs/sparse_rcnn/sparse_rcnn_r50_fpn_1x_coco.py
- configs/sparse_rcnn/sparse_rcnn_r50_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py 52 additions, 0 deletions...cnn_r50_fpn_300_proposals_crop_mstrain_480-800_3x_coco.py
- configs/sparse_rcnn/sparse_rcnn_r50_fpn_mstrain_480-800_3x_coco.py 23 additions, 0 deletions...parse_rcnn/sparse_rcnn_r50_fpn_mstrain_480-800_3x_coco.py
- mmdet/core/bbox/assigners/hungarian_assigner.py 2 additions, 2 deletionsmmdet/core/bbox/assigners/hungarian_assigner.py
- mmdet/models/dense_heads/__init__.py 2 additions, 1 deletionmmdet/models/dense_heads/__init__.py
- mmdet/models/dense_heads/embedding_rpn_head.py 120 additions, 0 deletionsmmdet/models/dense_heads/embedding_rpn_head.py
- mmdet/models/detectors/__init__.py 2 additions, 1 deletionmmdet/models/detectors/__init__.py
- mmdet/models/detectors/sparse_rcnn.py 110 additions, 0 deletionsmmdet/models/detectors/sparse_rcnn.py
- mmdet/models/losses/focal_loss.py 25 additions, 1 deletionmmdet/models/losses/focal_loss.py
- mmdet/models/roi_heads/__init__.py 2 additions, 1 deletionmmdet/models/roi_heads/__init__.py
- mmdet/models/roi_heads/bbox_heads/__init__.py 2 additions, 1 deletionmmdet/models/roi_heads/bbox_heads/__init__.py
- mmdet/models/roi_heads/bbox_heads/bbox_head.py 73 additions, 0 deletionsmmdet/models/roi_heads/bbox_heads/bbox_head.py
- mmdet/models/roi_heads/bbox_heads/dii_head.py 415 additions, 0 deletionsmmdet/models/roi_heads/bbox_heads/dii_head.py
- mmdet/models/roi_heads/sparse_roi_head.py 311 additions, 0 deletionsmmdet/models/roi_heads/sparse_roi_head.py
- mmdet/models/utils/__init__.py 2 additions, 2 deletionsmmdet/models/utils/__init__.py
- mmdet/models/utils/transformer.py 119 additions, 3 deletionsmmdet/models/utils/transformer.py
- tests/test_config.py 14 additions, 2 deletionstests/test_config.py
Loading
Please register or sign in to comment