[Refactor]: Support batch inference traceable by ONNX in SSD, YOLOv3, FSAF,...
[Refactor]: Support batch inference traceable by ONNX in SSD, YOLOv3, FSAF, RetinaNet, and FCOS (#4699)
* Support batch infer in RetinaNet
* Suport batch multiclass_nms
* Revert multiclass_nms
* Fix api deprecated warning
* do not repeat anchors
* Move img_shapes
* Update Yolov3
* Support FCOS
* Support RPN
* Fix RPN topk_inds error
* make batch exportable to onnx for yolohead
* make fcos_head exportable to onnx with batch dim
* Support ATSS
* Support CornerNet and centripetalNet
* Update RetinaNet and delta_xywh
* Remove ugly code
* Remove ugly code of FCOS
* Remove ugly code of ATSS/YOLOV3
* Support RPN and revert bbox_head
* expand anchors to batch and remove BG class when use deploy_nms_pre
* Update
* Use dim=-1 instead of dim=2
* Rename anchor_head method
* Keep the original format output when nms is not use
* Rename method and unified code style
* Fix paa_head and unittest
* Fix FASF onnx export error
* Fix error
* fix single stage img_shapes for onnx
* move conf_thr
* fix rpn_head for onnx
* Add distance2bbox unittest
* Remove TODO
* Fix RPN
* Update docstrs
Co-authored-by:
maningsheng <maningsheng@sensetime.com>
Showing
- configs/_base_/models/ssd300.py 1 addition, 0 deletionsconfigs/_base_/models/ssd300.py
- mmdet/core/bbox/coder/delta_xywh_bbox_coder.py 1 addition, 1 deletionmmdet/core/bbox/coder/delta_xywh_bbox_coder.py
- mmdet/core/bbox/coder/tblr_bbox_coder.py 4 additions, 3 deletionsmmdet/core/bbox/coder/tblr_bbox_coder.py
- mmdet/core/bbox/transforms.py 29 additions, 13 deletionsmmdet/core/bbox/transforms.py
- mmdet/models/dense_heads/anchor_head.py 109 additions, 74 deletionsmmdet/models/dense_heads/anchor_head.py
- mmdet/models/dense_heads/atss_head.py 114 additions, 76 deletionsmmdet/models/dense_heads/atss_head.py
- mmdet/models/dense_heads/cascade_rpn_head.py 131 additions, 1 deletionmmdet/models/dense_heads/cascade_rpn_head.py
- mmdet/models/dense_heads/dense_test_mixins.py 4 additions, 1 deletionmmdet/models/dense_heads/dense_test_mixins.py
- mmdet/models/dense_heads/fcos_head.py 112 additions, 69 deletionsmmdet/models/dense_heads/fcos_head.py
- mmdet/models/dense_heads/gfl_head.py 56 additions, 41 deletionsmmdet/models/dense_heads/gfl_head.py
- mmdet/models/dense_heads/paa_head.py 59 additions, 43 deletionsmmdet/models/dense_heads/paa_head.py
- mmdet/models/dense_heads/rpn_head.py 76 additions, 53 deletionsmmdet/models/dense_heads/rpn_head.py
- mmdet/models/dense_heads/yolo_head.py 120 additions, 86 deletionsmmdet/models/dense_heads/yolo_head.py
- tests/test_models/test_dense_heads/test_paa_head.py 5 additions, 5 deletionstests/test_models/test_dense_heads/test_paa_head.py
- tests/test_models/test_forward.py 1 addition, 1 deletiontests/test_models/test_forward.py
- tests/test_utils/test_misc.py 45 additions, 0 deletionstests/test_utils/test_misc.py
Loading
Please register or sign in to comment