mirrors/mmcv
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmcv.git synced 2025-06-03 21:54:52 +08:00
Code Issues Projects Releases Wiki Activity
1,421 Commits 13 Branches 108 Tags
Commit Graph

61 Commits

Author SHA1 Message Date
DaGaiBa
cd05d71254
Add multi_scale_deform_attn op adapter for NPU (#3032) 2024-03-01 15:02:23 +08:00
sherie
c7c02a7c5b
[Feature] torch_npu support aclnn and add op (#2998) 2024-01-07 17:11:36 +08:00
huaweiZJX
2e44eaeba3
[Feature] Add NPU operator RotatedFeatureAlign (#2994) 2023-11-29 17:24:28 +08:00
qipengh
86a38aa3f0
[Feature] Support modulated_deform_conv and deform_conv with cambricon MLU backend (#2823) 2023-07-27 16:28:42 +08:00
MrShadowY
987d34b0cf
[Feature] Add the support of BoxIouRotated op for ascend device (#2842) 
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
 2023-07-19 10:36:26 +08:00
Danielmic
d28aa8a9cc
[Feature] Add the implementation of diff_iou_rotated with mlu-ops (#2840) 2023-06-30 15:18:39 +08:00
Danielmic
10c8b9e78b
[Feature] Add the implementation of dynamic_scatter with mlu-ops (#2847) 2023-06-29 17:13:57 +08:00
tudejiang79
d2aecbe494
[Feature] Add the support for rotated_feature_align with MLU (#2809) 2023-06-01 00:55:20 +08:00
bdf
0c23eb0215
Sync main with mmcv1.x branch (#2800) 2023-05-19 15:40:12 +08:00
liuhw
9a5685a995
[Feature] Add the support of arf op for ascend device (#2792) 2023-05-10 17:07:17 +08:00
ckirchhoff
d493a5d523
Pick changes from 1.x branch (#2738) 2023-04-20 19:27:11 +08:00
bdf
733e6ff84e
Pick MLU modifications from master (1.x) to main (2.x) (#2704) 
* [Feature] Support Voxelization with cambricon MLU device (#2500)

* [Feature] Support hard_voxelize with cambricon MLU backend

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Feature](bangc-ops): add voxelization op

* [Enhance] Optimize the performace of ms_deform_attn for MLU device (#2510)

* ms_opt

* ms_opt

* ms_opt

* ms_opt

* ms_opt

* [Feature] ms_deform_attn performance optimization

* [Feature] ms_deform_attn performance optimization

* [Feature] ms_deform_attn performance optimization

* [Feature] Support ball_query with cambricon MLU backend and mlu-ops library. (#2520)

* [Feature] Support ball_query with cambricon MLU backend and mlu-ops library.

* [Fix] update operator data layout setting.

* [Fix] add cxx compile option to avoid symbol conflict.

* [Fix] fix lint errors.

* [Fix] update ops.md with info of ball_query support by MLU backend.

* [Feature] Fix typo.

* [Fix] Remove print.

* [Fix] get mlu-ops from MMCV_MLU_OPS_PATH env.

* [Fix] update MMCV_MLU_OPS_PATH check logic.

* [Fix] update error info when failed to download mlu-ops.

* [Fix] check mlu-ops version matching info in mmcv.

* [Fix] revise wrong filename.

* [Fix] remove f.close and re.

* [Docs] Steps to compile mmcv-full on MLU machine (#2571)

* [Docs] Steps to compile mmcv-full on MLU machine

* [Docs] Adjust paragraph order

* Update docs/zh_cn/get_started/build.md

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

* Update docs/zh_cn/get_started/build.md

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

* Update docs/en/get_started/build.md

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

* Update docs/en/get_started/build.md

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

* [Docs] Modify the format

---------

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

* [Fix] Fix tensor descriptor setting in MLU ball_query. (#2579)

* [Feature] Add MLU support for Sparse Convolution op (#2589)

* [Feature] Add sparse convolution MLU API

* [Feature] update cpp code style

* end-of-file

* delete libext.a

* code style

* update ops.md

---------

Co-authored-by: budefei <budefei@cambricon.com>

* [Enhancement] Replace the implementation of deform_roi_pool with mlu-ops (#2598)

* [Feature] Replace the implementation of deform_roi_pool with mlu-ops

* [Feature] Modify code

---------

Co-authored-by: budefei <budefei@cambricon.com>

* [Enhancement] ms_deform_attn performance optimization (#2616)

* ms_opt_v2

* ms_opt_v2_1

* optimize MultiScaleDeformableAttention ops for MLU

* ms_opt_v2_1

* [Feature] ms_deform_attn performance optimization V2

* [Feature] ms_deform_attn performance optimization V2

* [Feature] ms_deform_attn performance optimization V2

* [Feature] ms_deform_attn performance optimization V2

* [Feature] ms_deform_attn performance optimization V2

* [Feature] ms_deform_attn performance optimization V2

* [Feature] ms_deform_attn performance optimization V2

---------

Co-authored-by: dongchengwei <dongchengwei@cambricon.com>

* [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

* [Enhance] Add a default value for MMCV_MLU_ARGS (#2688)

* add mlu_args

* add mlu_args

* Modify the code

---------

Co-authored-by: budefei <budefei@cambricon.com>

* [Enhance] Ignore mlu-ops files (#2691)

Co-authored-by: budefei <budefei@cambricon.com>

---------

Co-authored-by: ZShaopeng <108382403+ZShaopeng@users.noreply.github.com>
Co-authored-by: BinZheng <38182684+Wickyzheng@users.noreply.github.com>
Co-authored-by: liuduanhui <103939338+DanieeelLiu@users.noreply.github.com>
Co-authored-by: budefei <budefei@cambricon.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: duzekun <108381389+duzekunKTH@users.noreply.github.com>
Co-authored-by: dongchengwei <dongchengwei@cambricon.com>
Co-authored-by: liuyuan1-v <125547457+liuyuan1-v@users.noreply.github.com>
 2023-04-19 10:42:07 +08:00
Zaida Zhou
463eab6b5d
[Docs] Describe branch changes (#2741) 
* [Docs] Describe branch changes

* fix typo

* Update English

* refine
 2023-04-06 23:42:14 +08:00
Yifei Yang
869dbf1bf2
[Feature] Add Ops of StyleGAN3 (#2290) 
* add bias_act

* support bias_act

* support filtered_lrelu

* support filtered_lrelu and upfirdn2d

* support conv2d_gradfix and fix filtered_lrelu

* fix lint

* fix lint

* fix c++ lint

* fix part comments

* fix lint

* rm redundant header

* fix upgrade pip

* fix as comment

* fix c++ lint

* fix ci

* fix-ut

* fix as comments

* add grad check

* remove redundant template

* Update mmcv/ops/bias_act.py

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

* add typehint

* fix as comment:

* complete type hints

* fix lint

* add test for conv_gradfix

* add test for conv_gradfix

* fix lint

* modify licenses and ops.md

* add zh op md

* add torch version policy for conv2d_gradfix

* fix lint

* fix as comments

* rename impl

* rm redudant function and add ut

* fix as comment

* fix lint

* fix lint

* fix as comments

* fix lint

* fix ut

* fix as comment

* fix as comment

* fix as comment

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
 2023-03-13 16:05:11 +08:00
liuhw
0fb07d0e69
[Feature] Add the support of voxelization op for ascend device (#2614) 
* Add voxelization op npu adaptor

* update

* update
 2023-03-03 20:07:11 +08:00
ckirchhoff
834f94db23
[Feature] Add support of some ops for Ascend device (#2594) 
* [Feature]: add supports of gather_points, nms_rotated, bbox_overlaps for Ascend device

* Apply suggestions from code review

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
 2023-02-13 16:43:42 +08:00
ckirchhoff
34bdf448ac
[Feature] Pick npu ops from master to 2.x (#2501) 
* merge npu ops from master to 2.x

* BugFix: fix merge bugs

* {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition

* merge conflicts in ops.md

* [fix]: fix merge bug
 2023-01-13 13:39:35 +08:00
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
liukuikun
fd11b874cf
[Docs] Add convention for the order of data transforms (#2442) 2022-12-06 17:06:20 +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
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
Xin Li
c9ed3f5245
[Docs] Translate data_transform.md to English (#2325) 
* draft 1 with some typo fixed

* draft 2

* Update docs/en/understand_mmcv/data_process.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>

* Remove timer tool

* Update docs/en/understand_mmcv/data_transform.md

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

* Update docs/en/understand_mmcv/data_transform.md

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

* Update docs/en/understand_mmcv/data_transform.md

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

* Update docs/en/understand_mmcv/data_transform.md

Co-authored-by: Qian Zhao <112053249+C1rN09@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
 2022-10-20 14:41:41 +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
Zaida Zhou
75ea2f8995
Upgrade the versions of pre-commit hooks (#2328) 2022-10-11 14:09:10 +08:00
yxzhao
c57b8b184b
[Docs] Fix some mistakes in data_process.md (#2283) 
* correction

* correction en docs
 2022-09-25 15:00:43 +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
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
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
Zaida Zhou
521d0b0e6c
[Fix] Fix error in version.py (#2133) 2022-07-19 20:06:09 +08:00
liyining
59eaefebe0 [Feature] Support partial mapping by manually marking keys as ignored 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
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
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
Ma Zerun
8b47579e7c [Docs] Add data transform docs. (#1723) 
* Add data transfrom docs

* Convert all Chinese paragraph in oneline to prevent some rendering bug
in Chrome/Edge. (It looks well in Firefox :/)

* Update docs

* Update docs

* Imporve docs according to comments

* Imporve docs according comments
 2022-07-19 19:18:15 +08:00
Reza
323abb9e5a
[Docs] Fix typos (#2124) 2022-07-17 18:41:17 +08:00
zhouchenyang
2708fac6c5
[Feature] Support RoiPool with Cambricon MLU backend (#2073) 
* [Feature] Support RoiPool with cambricon MLU backend

* [Docs] Update ops.md
 2022-06-29 11:24:00 +08:00
Alex Yang
9110df94fc
[Docs] Reformat op docs (#2052) 
* [Docs]:update ops.md

* fix missing ops
 2022-06-15 10:57:29 +08:00
Yue Zhou
834d5978df
[Feature] Add ChamferDistance op in gpu (#1933) 
* update

* fix lint

* fix lint

* Update mmcv/ops/chamfer_distance.py

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

* Update chamfer_distance.py

* Update test_chamfer_distance.py

* fix

* Update chamfer_distance_cuda_kernel.cuh

* Update chamfer_distance_cuda_kernel.cuh

* fix

* Update mmcv/ops/chamfer_distance.py

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

* Update chamfer_distance.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
 2022-06-08 18:12:57 +08:00
Danila Rukhovich
e9f48a4f8e
[Enhance] Replace BEV IoU with 3D IoU (#1902) 
* add iou3d

* revert deprecated python function

* fix lint

* replace 3d iou/nms calls for bev iou/nms
 2022-05-25 17:05:50 +08:00
Zaida Zhou
b326a219a7
[Docs] Replace markdownlint with mdformat for avoiding installing ruby (#1936) 
* Use mdformat pre-commit hook

* allows consecutive numbering

* improve .mdformat.toml

* test mdformat

* format markdown

* minor fix

* fix codespecll

* fix circleci

* add linkify-it-py dependency for cicleci

* add comments

* replace flake8 url

* add mdformat-myst dependency

* remove mdformat-myst dependency

* update contributing.md
 2022-05-16 20:47:56 +08:00
Wencheng Wu
a3a078eefb
[Enhancement] Support registering function (#1858) 
* [Enhance] Support register function.

* fix unittest error

* add docs and unittest of register function

* modify the docs

* fix version to 1.5.1

* Update docs/zh_cn/understand_mmcv/registry.md

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

* Update docs/en/understand_mmcv/registry.md

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

* refine the docs

* modify module_class to module

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
 2022-05-02 18:31:41 +08:00