q.yao
d8e4a78636
[Improvement] Better unit test. ( #1619 )
...
* update test for mmcls and mmdet
* update det3d mmedit mmocr mmpose mmrotate
* update mmseg
* bug fixing
* refactor ops
* rename variable
* remove comment
2023-02-08 11:30:59 +08:00
q.yao
1323ffcb50
[Refactor] Ease rewriter import ( #1166 )
...
* Ease rewriter import
* remove import
* add sar_encoder
* recover mha
2022-10-25 10:45:03 +08:00
q.yao
0e1a3aa8c7
[Feature] support MMRotate model with le135 ( #788 )
...
* support MMRotate model with le135
* cse before fuse select assign
* remove unused import
2022-07-25 16:01:05 +08:00
Hakjin Lee
36b3ca4461
[Feature] Support RoITransRoIHead ( #713 )
...
* [Feature] Support RoITransRoIHead
* Add docs
* Add mmrotate models regression test
* Add a draft for test code
* change the argument name
* fix test code
* fix minor change for not class agnostic case
* fix sample for test code
* fix sample for test code
* Add mmrotate in requirements
* Revert "Add mmrotate in requirements"
This reverts commit 043490075e
.
2022-07-22 15:19:37 +08:00
q.yao
dace58e844
[Enhancement] Support Slide Vertex TRT ( #650 )
...
* reorgnize mmrotate
* fix
* add hbb2obb
* add ut
* fix rotated nms
* update docs
* update benchmark
* update test
* remove ort regression test, remove comment
2022-07-13 16:09:09 +08:00
q.yao
0cac5154a6
[Enhancement] Support two-stage rotated detector TensorRT. ( #530 )
...
* upload
* add fake_multiclass_nms_rotated
* delete unused code
* align with pytorch
* Update delta_midpointoffset_rbbox_coder.py
* add trt rotated roi align
* add index feature in nms
* not good
* fix index
* add ut
* add benchmark
* move to csrc/mmdeploy
* update unit test
Co-authored-by: zytx121 <592267829@qq.com>
2022-06-23 10:35:10 +08:00
Yue Zhou
89204d16ce
[Feature] Support two stage rotated detector in MMRotate ( #473 )
...
* upload
* add fake_multiclass_nms_rotated
* delete unused code
* align with pytorch
* Update delta_midpointoffset_rbbox_coder.py
* rewrite atan2
* Update bbox_nms.py
2022-05-25 13:41:16 +08:00
Yue Zhou
42dc5bc316
Support single stage rotated detector in MMRotate ( #428 )
...
* fix lint
* fix lint
* add mmrotate part
* update
* update
* fix
* remove init_detector
* success run with bs=1
* nms_rotated support batch
* support [batch_id, class_id, box_id]
* fix
* fix
* Create test_mmrotate_core.py
* add ut
* add ut
* Update nms_rotated.py
* fix
* Revert "fix"
This reverts commit f792387fb4
.
* add mmrotate into requirements
* add ut
* update doc
* update
* skip test because mmcv version < 1.4.6
* update
* Update rotated-detection_static.py
* Update rotated-detection_static.py
* Update rotated-detection_static.py
* fix bug of memory leak.
* Update rotated_detection_model.py
2022-05-07 16:11:43 +08:00