ba8aa76458
[Refactor] Repalce the implementation of roi_align_rotated with mlu-ops ( #2808 )
2023-06-01 00:56:49 +08:00
d2aecbe494
[Feature] Add the support for rotated_feature_align with MLU ( #2809 )
2023-06-01 00:55:20 +08:00
7ff7095ca7
[Fix] Fix the support for ROCm ( #2811 )
2023-05-31 12:47:46 +08:00
0c23eb0215
Sync main with mmcv1.x branch ( #2800 )
2023-05-19 15:40:12 +08:00
9a5685a995
[Feature] Add the support of arf op for ascend device ( #2792 )
2023-05-10 17:07:17 +08:00
d493a5d523
Pick changes from 1.x branch ( #2738 )
2023-04-20 19:27:11 +08:00
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
1db3967e86
Fix lint ( #2702 )
2023-03-26 02:06:28 +08:00
5b0cdccc5c
[Fix] Enhance the compatibility of training stylegan 2 ( #2694 )
...
* [Fix] Enhance the compatibility of training stylegan 2
* Fix unit test in PyTorch 2.0
* Apply suggestions from code review
---------
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-26 01:58:38 +08:00
a6d7dde77f
[Fix] Fix torch2.0 dcn/mdcn symbolic ( #2695 )
...
* fix
* fix lint
2023-03-24 13:01:56 +08:00
4ad79591f9
[Enhancement] Support MultiScaleDeformableAttention with AMP ( #2541 )
...
* [Enhance] Support FP16 for MSDeformAttn
* [Fix] Data type mismatch
* Update mmcv/ops/multi_scale_deform_attn.py
* Add UT
Author: nijkah <nijkah@gmail.com>
* Add cuda available condition
---------
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-20 18:26:52 +08:00
0e14ce2c1d
[Feature] Add C++ implementation for bbox_overlaps ( #2477 )
...
* add ops bbox_overlaps
* format code
* Return the pytorch version
* Intermediate modification
* Solve problems in parameter passing
* revise bug
* "add test case"
2023-03-20 18:26:52 +08:00
e7adffb9e0
[Fix] Skip filtered_lrelu ut when cuda is less than 10.2 ( #2677 )
2023-03-19 16:17:45 +08:00
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
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
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
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
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
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
a0939977a3
[Feature] Support MultiScaleDeformableAttn with cambricon MLU backend
2022-11-23 16:57:35 +08:00
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
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
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
3bb0611aef
[Fix] Create Tensor with new_* method to support amp ( #2389 )
2022-11-23 16:57:35 +08:00
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
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
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
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
e0b3223baf
fix ci
2022-10-22 13:39:50 +08:00
cb2eb57683
Remove onnx ( #2295 )
...
* remove onnx
* remove print
2022-10-21 15:35:07 +08:00
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
cdfbdc0bd1
[Feature] Support ThreeNN with cambricon MLU backend ( #2215 )
2022-08-28 17:54:20 +08:00
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
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
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
c3835415ed
[Refactor] Modify the importing path due to mmengine changes ( #2224 )
2022-08-24 19:48:10 +08:00
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
0fe1c64744
Remove fileio from mmcv and use mmengine.fileio instead ( #2179 )
2022-08-16 13:37:19 +08:00
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
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
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
15495ea0a7
[Fix] make iou3d.boxes_iou3d actually calculate 3D IoU ( #2018 )
...
* fix iou3d
* rename variables in test
* fix comments
2022-06-09 21:08:32 +08:00
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
5601427bbc
[Feature] Support PSAMask with cambricon MLU backend ( #2024 )
...
* [Feature] Support Psamask with cambricon MLU backend
* [Fix] refine the format according to lint check
* Replace std:max/min with conditional operators
* [Fix] Format head files
2022-06-08 16:13:13 +08:00
fb5062ca00
[Feature] Support RoiAlignRotated with cambricon MLU backend ( #2033 )
...
* [Feature] Support RoiAlignRotated with cambricon MLU backend
* [Fix] Remove std lib in mlu files
* [Fix] replace std::min/max with conditional operators
2022-06-08 14:13:28 +08:00
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
7e6f462495
[Fix] Skip CPU test in test_rotated_feature_align.py for parrots ( #1968 )
...
* skip test_rotated_feature_align.py for parrots
Signed-off-by: del-zhenwu <dele.zhenwu@gmail.com>
* fix lint
Co-authored-by: zhouzaida <zhouzaida@163.com>
2022-05-20 17:04:58 +08:00
60eadb0665
[Enhancement] Lower the restrictions of _resize method in BaseMergeCell ( #1959 )
...
* Fix the bug met in using nasfpn
Fix the bug met in using nasfpn which is mentioned at https://github.com/open-mmlab/mmdetection/issues/5987 .
Avoid the strong restrictions of _resize function in BaseMergeCell:
1. When Downsampling the feature map, the feature map's shape must be divisible by the target size. We pad zero around feature map before max_pool2d opt to make it always divisible. (line 102 ~ 107)
2. Considering the different downsampling scale of H and W, shape[-2] and shape[-1] are involed in the definition of kernel_size. (line 110)
* Update merge_cells.py
check flake8 & isort
* Update merge_cells.py
* Update merge_cells.py
yapf
* Update mmcv/ops/merge_cells.py
X_pad rename to padding_x
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
* Update merge_cells.py
format the code style after renameing the X_pad to padding_x
* Update test_merge_cells.py
Mainly test the downsampling resize in BaseMergeCell. The smaller target size is set to (14, 7), the classical feature map's size in the last few stages of the backbone, which will product different downsampling scales in different dims.
* Update test_merge_cells.py
add "# Copyright (c) OpenMMLab. All rights reserved."
* Update merge_cells.py
format the variable name
* Update test_merge_cells.py
Testing divisible and indivisible situations simultaneously
* Update mmcv/ops/merge_cells.py
fix the bug when h is indivisible and w is divisible, the pad_w will be padded unreasonable.
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/ops/merge_cells.py
fix the bug when w is indivisible and h is divisible, the pad_h will be padded unreasonable.
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* fix undefined error
* Update merge_cells.py
make pad_h, pad_w more readable
* Update test_merge_cells.py
use @pytest.mark.parametrize instead of 'for' methor
* Update merge_cells.py
* Update test_merge_cells.py
isort
* Update merge_cells.py
isort
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-20 15:13:13 +08:00
45fa3e44a2
Add pyupgrade pre-commit hook ( #1937 )
...
* add pyupgrade
* add options for pyupgrade
* minor refinement
2022-05-18 11:47:14 +08:00
e70a318ae9
[Fix] Fix parrots unit tests ( #1913 )
2022-05-09 16:59:32 +08:00
tudejiang79
xiabo123
bdf
liuhw
ckirchhoff
Zaida Zhou
Mashiro
q.yao
Hakjin Lee
enemy1205
Yifei Yang
liukuikun
takuoko
ZShaopeng
cathyzhang222
Zhang
VVsssssk
CokeDong
Yue Zhou
liuduanhui
zhouchenyang
Yanyi Liu
Danila Rukhovich
del-zhenwu
JarvisKevin
pc