3 Commits

Author	SHA1	Message	Date
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 f792387fb449ba091c1d932f29d28214805fb6e3. * 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