Commit Graph

8 Commits (5234915c5efda6d080223be6ff6606874876e901)

Author SHA1 Message Date
liuyuan1-v 0d1b224fb1
[Feature] Support NmsRotated with cambricon MLU backend (#2643)
* [Feature] Support NmsRotated with cambricon MLU backend

* [Feature] remove foolproofs in nms_rotated_mlu.cpp

* [Feature] fix lint in test_nms_rotated.py

* [Feature] fix kMLU not found in nms_rotated.cpp

* [Feature] modify mlu support in nms.py

* [Feature] modify nms_rotated support in ops.md

* [Feature] modify ops/nms.py
2023-03-23 11:36:32 +08:00
ckirchhoff 54ed0ed869
[Feature] Add support for Ascend devices with nms_rotated (#2550)
* [Feature]: add nms_rotated npu adaptater code

* [BugFix]: modify param in nms_rotated_npu.cpp

* [clean code]: nms_rotated_npu.cpp

* [clean code]: nms_rotated_npu.cpp

* [clean code]: nms_rotated_npu.cpp

* [clean code]: nms_rotated.cpp

* [Doc]: add nms_rotated op in supported op list at ops.md

* [Test]: add nms_rotated unit_test

* [Bug]: remove device parameter in test_batched_nms function
2023-01-28 21:08:22 +08:00
Yanyi Liu 9807c2d268
[Fix] Fix batched_nms for rotated box and add type hints for nms.py (#2006)
* Fix batched_nms for rotated box.
Add type hint for nms.py

* Add test

* doc string

* revert symbolic hint

* fix max_coordinate

* add comment

* rename type

* fix typo docstring
2022-06-10 19:49:07 +08:00
Zaida Zhou 6e9ce18323
Add copyright pre-commit-hook (#1742)
* first commit

* Add copyright pre-commit-hook
2022-02-24 09:24:25 +08:00
Hakjin Lee b10eaedd38
[Fix] batched nms for nms_rotated (#1688)
* Fix batched_nms to support nms_rotated

* Update test code for batched_nms with nms_rotated

* Edit indexing the score in nms

* Use -1 indexing for batched_nms
2022-02-12 14:45:39 +08:00
zhanggefan b6167d5987
[Enhancement] Add clockwise argument to ops box_iou_rotated and nms_rotated (#1592)
* add clockwise arguments to ops box_iou_rotated and nms_rotated

* refactor docs

* change code that may incur stopped gradient.

* refactor docs

* Update mmcv/ops/nms.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

* Update mmcv/ops/box_iou_rotated.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-01-10 10:57:11 +08:00
WRH f0e68404d2
Fix some bug as well as unit test for nms_rotate (#728)
* add const to nms_rorate in pybind

* fix test nms rotated

* skip test instead of passing it

* fix lint

* update pytest skip syntax
2020-12-18 15:17:11 +08:00
BigBigDream a978764883
Add box_iou_rotated, ml_nms_rotated and nms_rotated (#625)
* add box_iou_rotated, ml_nms_rotated and nms_rotated

* fix lint

* fix lint

* fix .py lint

* fix cpp lint

* add newline at the end

* add new line

* fix unittest

* config google style

* fix lint

* lint

* lint

* yapf

* update

* fix lint

* fix lint

* fix lint

* fix

* fix format

* fix format

* add modified from

* add docstring and update others

* update docstring

* update docstring

* update

* fix bug

* fix bug

* fix bug

Co-authored-by: Cao Yuhang <yhcao6@gmail.com>
2020-11-25 11:00:30 +08:00