Commit Graph

469 Commits (7156604eba7350bb6c43f29d0fb2ee336f8d9dff)

Author SHA1 Message Date
liukuikun 7156604eba
[Feature] Add BezierAlign CUDA op (#2393)
* bezier align

* add ut

* fix comment

* updata ut

* fix link and comment

* fix comment
2023-01-10 17:05:39 +08:00
Zaida Zhou a3cb068590
[Refactor] Use the new interface of fileio from mmengine (#2468)
* [Refactor] Use new interface of fileio

* update ut

* remove file_client_args

* add * before backend_args

* refine backend_args description
2022-12-28 00:09:05 +08:00
takuoko d510b8b174
[Feature] Support LayerScale in FFN (#2451)
* add layer scale

* add layer scale

* add layer scale

* Update mmcv/cnn/bricks/transformer.py

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

* Update mmcv/cnn/bricks/transformer.py

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

* add layer scale

* move LayerScale

* add layer_scale_init_value

* add typehint

* fix for tensor with any dim

* fix layer scale rule

* fix layer scale rule

* fix test

* add docs

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-12-11 17:58:19 +08:00
takuoko 433607030a
[Enhancement] Support value_proj_ratio in MultiScaleDeformableAttention (#2452)
* add ratio in ms_deform_attn_

* add ratio in ms_deform_attn

* Update mmcv/ops/multi_scale_deform_attn.py

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

* Update tests/test_ops/test_ms_deformable_attn.py

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

* add ratio in ms_deform_attn

* add ratio in ms_deform_attn

* add ratio in ms_deform_attn

* add ratio in ms_deform_attn

* add ratio in ms_deform_attn

* add ratio in ms_deform_attn

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-12-11 17:48:21 +08:00
ZShaopeng a0939977a3 [Feature] Support MultiScaleDeformableAttn with cambricon MLU backend 2022-11-23 16:57:35 +08:00
cathyzhang222 6c89b71756 [Feature] Support Iou3d with cambricon MLU backend (#2339)
* [Feature] Support Iou3d with cambricon MLU backend

* [Feature] add double line

* [Feature] add line
2022-11-23 16:57:35 +08:00
Zhang e9777adbac [Feature] Support RoIAwarePool3d with cambricon MLU backend (#2359)
* [Feature] Support RoiawarePool3d with cambricon MLU backend.

* [Feature] Support RoiawarePool3d with cambricon MLU backend.
2022-11-23 16:57:35 +08:00
q.yao 2d10616b8f [Fix] Fix three nn op can not accept half tensor (#2348)
* Fix three nn half inpt

* update test
2022-11-23 16:57:35 +08:00
q.yao 3bb0611aef [Fix] Create Tensor with new_* method to support amp (#2389) 2022-11-23 16:57:35 +08:00
VVsssssk 4e0341f26f
[Features] Add stack ball query and stack group points ops (#2292)
* add stack sa model ops

* fix lint

* fix lint

* fix comments

* fix bug

* fix lint

* fix comments

* fix lint

* fix lint

* fix
2022-10-28 00:42:10 +08:00
ZShaopeng a8f7ae48e2 [Feature] Support RoipointPool3d with cambricon MLU backend (#2247)
* [Feature] Support RoipointPool3d with cambricon MLU backend

* [Feature] Support RoipointPool3d with cambricon MLU backend

* [Feature] Support RoipointPool3d with cambricon MLU backend

* [Feature] Support RoipointPool3d with cambricon MLU backend

* [Feature] Support RoipointPool3d with cambricon MLU backend
2022-10-22 13:39:50 +08:00
bdf a364e6cad2 [Feature] Support DeformRoiPool with cambricon MLU backend (#2137)
* [Feature] Support DeformRoiPool with cambricon MLU backend

* [Fix] Remove use of std library

* [Fix] Correct the error information

* [Refactor] Refactor test deform_roi_pool code

* [Fix] Fix judgment error

* [Fix] Modify the large tensor check

Co-authored-by: budefei <budefei@cambricon.com>
2022-10-22 13:39:50 +08:00
CokeDong 8fae3b4682 [Feature] Support PrRoI op for Parrots (#2280)
* Support parrots extension for op PrRoI

* Fix lint

* Fix cpp lint

* Fix testcase failure by false requires_grad in self-defined autograd Funtion

* Fix issues

* Fix flake8

* Fix isort

* Adaption for typechecking for PrRoIPoolFunction

* Fix lint

* Support only float32

* bugfix
2022-10-22 13:39:50 +08:00
takuoko d8ac46df59 [Enhancement] Support SiLU with torch < 1.7.0 2022-10-22 13:39:50 +08:00
takuoko dfef152989 [Feature] Register a new activatation layer SiLU to ACTIVATION_LAYERS 2022-10-22 13:39:50 +08:00
q.yao e0b3223baf fix ci 2022-10-22 13:39:50 +08:00
q.yao cb2eb57683
Remove onnx (#2295)
* remove onnx

* remove print
2022-10-21 15:35:07 +08:00
Mashiro a6c42ad324
[Feature] Add TTA transform (#2146)
* Add TestRandomResize

* rename ut class

* minor retine

* Replace for loop with itertools.product

* Support accept built transforms

* Fix unit test

* Refine docstring

* minor refine

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

* Minor refine

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-10-19 23:32:22 +08:00
Miao Zheng bf48ca035b
[Fix] Add `swap_labe_pairs` in `RandomFlip` (#2332)
* [Fix] Add `swap_labe_pairs` in `RandomFlip`

* [Fix] Add `swap_labe_pairs` in `RandomFlip`

* add reference info

* add swap_label_pairs in results

* revise according to comments

* revise according to comments

* revise according to comments

* docstring

* docstring
2022-10-17 23:21:20 +08:00
Yue Zhou a4c82617ff
[Feature] Add box_iou_quadri & nms_quadri (#2277)
* update

* update

* fix link

* fix bug

* update nms_quadri

* fix lint

* Update test_nms_quadri.py

* Update box_iou_quadri.py

* fix bug

* Update test_nms_quadri.py

* Update box_iou_rotated_utils.hpp

* Update box_iou_quadri.py

* Update mmcv/ops/nms.py
2022-10-13 17:26:14 +08:00
liuduanhui cdfbdc0bd1 [Feature] Support ThreeNN with cambricon MLU backend (#2215) 2022-08-28 17:54:20 +08:00
bdf 625e82ce62 [Feature] Add carafe op for MLU (#2212)
* [Feature] Support CARAFE with Cambricon MLU backend

* [Docs] Add comments for common funtions

* [Test] Add allclose test for carafe

* Remove print

Co-authored-by: zcyKTH <zcy19950525@gmail.com>
Co-authored-by: budefei <budefei@cambricon.com>
2022-08-28 17:54:20 +08:00
bdf f5a19ef06f [Feature] Support MaskedConv2d with cambricon MLU backend (#2202)
* [Feature] Support MaskedConv2d with cambricon MLU backend

* [Refactor] Refactor test masked_conv2d code

Co-authored-by: budefei <budefei@cambricon.com>
2022-08-28 17:54:20 +08:00
q.yao 2e5628b410
[Refactor]: Remove deployment for dev-2.x (#2225)
* remove deploy for 2.0

* update onnx ut
2022-08-26 20:11:05 +08:00
Zaida Zhou c3835415ed
[Refactor] Modify the importing path due to mmengine changes (#2224) 2022-08-24 19:48:10 +08:00
Zaida Zhou 5947178e85
Remove many functions in utils and migrate them to mmengine (#2217)
* Remove runner, parallel, engine and device

* fix format

* remove outdated docs

* migrate many functions to mmengine

* remove sync_bn.py
2022-08-23 10:12:41 +08:00
Zaida Zhou 9185eee858
Remove runner, parallel, engine and device (#2216)
* Remove runner, parallel, engine and device

* fix format

* remove outdated docs
2022-08-22 19:16:30 +08:00
Mashiro 19a024155a
[Refactor] Use MODELS registry in mmengine and delete basemodule (#2172)
* change MODELS to mmengine, delete basemodule

* fix unit test

* remove build from cfg

* add comment and rename TARGET_MODELS to registry

* refine cnn docs

* remove unnecessary check

* refine as comment

* refine build_xxx_conv error message

* fix lint

* fix import registry from mmcv

* remove unused file
2022-08-19 11:18:47 +08:00
Jingwei Zhang 86f9dc7a40 [Enhance] Support different border modes in imrotate (#2141)
* support different border_mode in imrotate

* fix docstring

* small fix

* reorder the border_mode arg in imrotate

* fix lint

* polish docstring

* polish docstring
2022-08-18 19:20:57 +08:00
Zaida Zhou 0fe1c64744
Remove fileio from mmcv and use mmengine.fileio instead (#2179) 2022-08-16 13:37:19 +08:00
Zaida Zhou 0b4285d959
Pick commits from master (#2164)
* [Docs] Add swith_language.md in docs (#2160)

* [Fix] Fix onnx unit tests (#2155)

* [Docs] Limit extension versions (#2144)

* Support PrRoIPool operation

* Add MPS bbox overlap

* Add .pre-commit-config-zh-cn.yaml (#2135)

Co-authored-by: xcnick <xcnick0412@gmail.com>
Co-authored-by: Jingwei Zhang <zjw18@mails.tsinghua.edu.cn>
Co-authored-by: q.yao <yaoqian@sensetime.com>
2022-08-03 14:11:53 +08:00
Yixiao Fang e5c2cab583 [Refactor] Add pillow backend for ColorJitter related functions (#2127)
* add 'backend' for ColorJitter related functions

* add unittest

* fix unittest

* add bgr transpose and revise ut

* update unittest

* revise docstring
2022-07-19 19:18:16 +08:00
plyfager 1a57a76be5 complete repr functions 2022-07-19 19:18:16 +08:00
liukuikun f90567a091 [Fix] LoadImageFromFile 2022-07-19 19:18:16 +08:00
huanghaian 864942be2f fix resize error 2022-07-19 19:18:16 +08:00
Ma Zerun a1fb002269 [Refactor] Use `resize_type` instead of `resize_cfg` in `RandomResize` and `RandomChoiceResize`. 2022-07-19 19:18:16 +08:00
YuanLiuuuuuu a03774d6db [Refactor]: Grascale return uint8 type 2022-07-19 19:18:16 +08:00
YuanLiuuuuuu 5867a97a41 [Fix]: Fix random resize 2022-07-19 19:18:16 +08:00
liyining 59eaefebe0 [Feature] Support partial mapping by manually marking keys as ignored 2022-07-19 19:18:16 +08:00
gongtao.vendor 3b494a1304 Support broadcasting all keys for TransformBroadcaster 2022-07-19 19:18:16 +08:00
liukuikun 88f3cc3f35 replace height, width with img_shape 2022-07-19 19:18:16 +08:00
liukuikun ab0aa4ef86 load type (#1921) 2022-07-19 19:18:16 +08:00
Yining Li f59aec8ffb [Feature] Add decorator `avoid_cache_randomness` (#1864)
* add prohibit_cache_randomness

* rename as avoid_cache_randomness and ensure it is non-inheritable

* fix lint

* update docs
2022-07-19 19:18:16 +08:00
Yifei Yang ea84b67449 update scales and img_shape (#1871) 2022-07-19 19:18:16 +08:00
Yuan Liu 30b3844621 [Fix]: Update img_shape in resize related transform (#1907) 2022-07-19 19:18:16 +08:00
Yining Li ff0dfb74d7 add RandomApply (#1863) 2022-07-19 19:18:16 +08:00
Ma Zerun 428512f88a Rename `LoadAnnotation` to `LoadAnnotations` (#1850) 2022-07-19 19:18:16 +08:00
Yifei Yang b7525fae1e [Fix] Fix typos, init funcs, cacheable method of part3 of data transforms (#1784)
* fix typos and move args into cfg

* update docstring

* fix as comment

* fix lint

* update transforms as discussed before

* linting

* fix as comment

* fix lint

* fix lint and also update according to PackInput

* remove precommit change

* rename cacheable method

Co-authored-by: liukuikun <641417025@qq.com>
2022-07-19 19:18:16 +08:00
liukuikun e7592a70f2 [Fix] fix some typo in transforms (#1831)
* [Fix] fix some typo according to review

* fix comment
2022-07-19 19:18:16 +08:00
Yining Li e2ca0733c6 Refactor base transform (#1830)
* rename cacheable_method to cache_randomness

* refactor transform wrappers and update docs

* fix all_nonexist_keys

* fix lint

* rename transform wrappers
2022-07-19 19:18:16 +08:00