73856344f5
[Feature] Support PSAMask op for Ascend device ( #2487 )
2023-01-06 20:15:50 +08:00
935ba78b39
[Fix] Fix skip_layer for RF-Next ( #2489 )
...
* judge skip_layer by fullname
* lint
* skip_layer first
* update unit test
2022-12-28 15:27:34 +08:00
46eb9ec5d0
[Feature] Add NPU adapter for Deformable RoIPool ( #2481 )
...
* add npu adapter for deformroipool
* cleancode
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
* Update deform_roi_pool.cpp
2022-12-14 15:09:46 +08:00
4c51afce2a
[Feature] Add FusedBiasLeakyRelu npu adapter ( #2474 )
...
* init npu
* add npu extension and focal loss adapter
* clean code
* clean code
* clean code
* clean code
* fix autocast bugs on npu (#2273 )
fix autocast bugs on npu (#2273 )
* code format
* code format
* code format
* bug fix
* pytorch_npu_helper.hpp clean code
* Npu dev (#2306 )
* fix autocast bugs on npu
* using scatter_kwargs in mmcv.device.scatter_gather
* raise ImportError when compile with npu
* add npu test case (#2307 )
* add npu test case
* Update focal_loss.py
* add comment
* clean lint
* update dtype assert
* update DDP forward and comment
* fix bug
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* sigmoidfocalloss npu adapter bug fix
* BugFix: modify softmaxFocalLoss adapter
* BugFix: remove equal sign in the code
* add npu install information in README
* add modulatedDeformConv npu adapter
* init npu
* add npu extension and focal loss adapter
* clean code
* clean code
* clean code
* add modulatedDeformConv npu adapter
* merge master branch 20221103
* Add masked_ Conv2d operator in NPU
* add nms_npu
* fix bug
* fix code check
* fix code check
* fix code check
* Masked_conv2d NPU
* Masked_conv2d NPU
* Masked_conv2d NPU
* remove npu-install-info in README.md
* annotate the clang-format in pre-commit-config-zh-ch.yaml
* Clean code: fix the clean code problem in masked_conv2d and modulated_deform_conv
* Create fused_bias_leakyrelu_npu.cpp
Add NPU adapter for fused_bias_leaky_relu operator
* Update fused_bias_leakyrelu_npu.cpp
* Update fused_bias_leakyrelu_npu.cpp
* Update ops.md
* Update ops.md
* Update fused_bias_leakyrelu_npu.cpp
* Update fused_bias_leakyrelu_npu.cpp
* Update test_fused_bias_leakyrelu.py
* Update fused_bias_leakyrelu.py
* Update test_fused_bias_leakyrelu.py
* Update fused_bias_leakyrelu.py
* Update test_fused_bias_leakyrelu.py
* Update ops.md
* amend for CI
* bugfix
* amend ops.md
* Update test_fused_bias_leakyrelu.py
* clean code
* bugfix
* clean code
* Update fused_bias_leakyrelu_npu.cpp
* Update fused_bias_leakyrelu_npu.cpp
Co-authored-by: wangjiangben <wangjiangben_hw@126.com>
Co-authored-by: ckirchhoff <515629648@qq.com>
Co-authored-by: wangjiangben-hw <111729245+wangjiangben-hw@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: zcc-zjut <zcczxy2019@163.com>
Co-authored-by: wangxiaoxin_sherie <wangxiaoxin7@huawei.com>
Co-authored-by: momo609 <963372609@qq.com>
2022-12-13 11:13:46 +08:00
4e67e481a0
[Fix] Fix ut caused by the new version of packaging ( #2482 )
2022-12-11 17:59:11 +08:00
49d891754b
[Feature] Support modulated_deform_conv with cambricon MLU backend ( #2411 )
...
* [Feature] Support modulated_deform_conv with cambricon MLU backend
* fix error of torch_mlu
* modify with commit suggestion
* Update modulated_deform_conv.py
* Update mmcv/ops/modulated_deform_conv.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-12-04 17:46:44 +08:00
c1712ee290
[Enhance] Add conf_path in PetrelBackend ( #2459 )
...
* feat: add conf_path in PetrelBackend
* fix: add conf_path in MockPetrelClient
* fix: adjust parameters order
2022-11-28 11:48:13 +08:00
6e3827d02b
[Feature] Support receptive field search of CNN models ( #2056 )
...
* support rfsearch
* add labs for rfsearch
* format
* format
* add docstring and type hints
* clean code
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* rm unused func
* update code
* update code
* update code
* update details
* fix details
* support asymmetric kernel
* support asymmetric kernel
* Apply suggestions from code review
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Apply suggestions from code review
* Apply suggestions from code review
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Apply suggestions from code review
* Apply suggestions from code review
* Apply suggestions from code review
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Apply suggestions from code review
* add unit tests for rfsearch
* set device for Conv2dRFSearchOp
* Apply suggestions from code review
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* remove unused function search_estimate_only
* move unit tests
* Update tests/test_cnn/test_rfsearch/test_operator.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Apply suggestions from code review
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/cnn/rfsearch/operator.py
Co-authored-by: Yue Zhou <592267829@qq.com>
* change logger
* Update mmcv/cnn/rfsearch/operator.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: lzyhha <819814373@qq.com>
Co-authored-by: Zhongyu Li <44114862+lzyhha@users.noreply.github.com>
Co-authored-by: Yue Zhou <592267829@qq.com>
2022-11-22 19:15:55 +08:00
12442667ff
[Feature] Support MultiScaleDeformableAttn with cambricon MLU backend ( #2396 )
...
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support MsDeformAttnForward with cambricon MLU backend
* [Feature] Support ms with cambricon MLU backend
* [Feature] Support msdeformattn_1104 with cambricon MLU backend
* [Feature] Support ms with cambricon MLU backend
* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend
* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend
* [Feature] Support MsDeformAttn_1108 with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* Revert "[Feature] Support MsDeformAttn with cambricon MLU backend"
This reverts commit 27963cccc86d240852a40a2c1510147a3e9f269f.
* [Feature] Support MsdeformAttn with cambricon MLU backend
* [Feature] Support MsDeformAttn with cambricon MLU backend
* [Feature] Support MsdeformAttn with cambricon MLU backend
Co-authored-by: zhangshaopeng <wicky-zheng@outlook.com>
Co-authored-by: wicky-zheng <root@notebook-mmcv-290m8-torch1-6-y0yv66-notebook-0.notebook-mmcv-290m8-torch1-6-y0yv66.ns-ad8b689a0ecd41fcb4469c803dcd539d.svc.cluster.local>
2022-11-16 14:08:04 +08:00
de6b0021af
[Feature] Support masked_conv in Ascend device ( #2387 )
...
* init npu
* add npu extension and focal loss adapter
* clean code
* clean code
* clean code
* clean code
* fix autocast bugs on npu (#2273 )
fix autocast bugs on npu (#2273 )
* code format
* code format
* code format
* bug fix
* pytorch_npu_helper.hpp clean code
* Npu dev (#2306 )
* fix autocast bugs on npu
* using scatter_kwargs in mmcv.device.scatter_gather
* raise ImportError when compile with npu
* add npu test case (#2307 )
* add npu test case
* Update focal_loss.py
* add comment
* clean lint
* update dtype assert
* update DDP forward and comment
* fix bug
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* sigmoidfocalloss npu adapter bug fix
* BugFix: modify softmaxFocalLoss adapter
* BugFix: remove equal sign in the code
* add npu install information in README
* add modulatedDeformConv npu adapter
* init npu
* add npu extension and focal loss adapter
* clean code
* clean code
* clean code
* add modulatedDeformConv npu adapter
* merge master branch 20221103
* Add masked_ Conv2d operator in NPU
* add nms_npu
* fix bug
* fix code check
* fix code check
* fix code check
* Masked_conv2d NPU
* Masked_conv2d NPU
* Masked_conv2d NPU
* remove npu-install-info in README.md
* annotate the clang-format in pre-commit-config-zh-ch.yaml
* Clean code: fix the clean code problem in masked_conv2d and modulated_deform_conv
Co-authored-by: wangjiangben <wangjiangben_hw@126.com>
Co-authored-by: wangjiangben-hw <111729245+wangjiangben-hw@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: zcc-zjut <zcczxy2019@163.com>
Co-authored-by: wangxiaoxin_sherie <wangxiaoxin7@huawei.com>
Co-authored-by: momo609 <963372609@qq.com>
2022-11-15 10:23:17 +08:00
6254ebef8d
[Fix] Create Tensor with new_* method to support amp ( #2389 )
2022-11-08 15:54:34 +08:00
b622fb2e29
[Feature] Support RoIAwarePool3d with cambricon MLU backend ( #2359 )
...
* [Feature] Support RoiawarePool3d with cambricon MLU backend.
* [Feature] Support RoiawarePool3d with cambricon MLU backend.
2022-11-03 11:13:34 +08:00
93fe4829f7
[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 16:53:33 +08:00
a4f304a5f5
[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-28 16:53:33 +08:00
9709ff3f8c
[Enhancement] Add a new argument define_metric in wandb hook ( #2237 )
...
* wandb define_metric
* add test and some fix based on mmengine PR
* fix test
* add summary warnings
* Update mmcv/runner/hooks/logger/wandb.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
* Update mmcv/runner/hooks/logger/wandb.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-10-24 23:13:07 +08:00
7fd7058a9c
[Feature] Support Iou3d with cambricon MLU backend ( #2339 )
...
* [Feature] Support Iou3d with cambricon MLU backend
* [Feature] add double line
* [Feature] add line
2022-10-24 15:58:58 +08:00
6cf9ac0507
[Fix] Fix three nn op can not accept half tensor ( #2348 )
...
* Fix three nn half inpt
* update test
2022-10-24 11:15:21 +08:00
e417035f5d
[Enhancement] Add ability to pass logger instance to frameworks ( #2317 )
...
* Add ability to pass logger instance to frameworks
* refine docstring
* Update mmcv/runner/hooks/logger/dvclive.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-10-12 12:09:53 +08:00
dd5b415d2a
[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-11 15:21:00 +08:00
c001e2fcba
[Refactor] Refactor the nms3d op to support MLU ( #2296 )
...
* update iou3d
* fix parrot
* update to device
* replace count_nonzero with fill
* update build.yml
2022-10-11 10:24:46 +08:00
b6a7fd98e4
Upgrade pre commit hooks ( #2321 )
...
* Upgrade the versions of pre-commit hooks
* update the versions of zh-cn.yaml
2022-10-08 11:48:44 +08:00
a5db5f6682
[Feature] Support training on NPU device ( #2262 )
...
* init npu
* add npu extension and focal loss adapter
* clean code
* clean code
* clean code
* clean code
* fix autocast bugs on npu (#2273 )
fix autocast bugs on npu (#2273 )
* code format
* code format
* code format
* bug fix
* pytorch_npu_helper.hpp clean code
* Npu dev (#2306 )
* fix autocast bugs on npu
* using scatter_kwargs in mmcv.device.scatter_gather
* raise ImportError when compile with npu
* add npu test case (#2307 )
* add npu test case
* Update focal_loss.py
* add comment
* clean lint
* update dtype assert
* update DDP forward and comment
* fix bug
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: ckirchhoff <515629648@qq.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-09-30 21:05:37 +08:00
afff388692
[Enhancement] Support SiLU with torch < 1.7.0 ( #2278 )
...
* support silu torch<1.7.0
* fix test
* fix test
* fix inplace
* fix inplace
* Update tests/test_cnn/test_silu.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-09-29 15:19:41 +08:00
55c51e1734
[Feature] Register a new activatation layer SiLU to ACTIVATION_LAYERS ( #2256 )
2022-09-13 22:38:23 +08:00
83bedf5a2b
[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-09-08 17:19:12 +08:00
e843d73fc7
[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-08-28 22:37:28 +08:00
a08517790d
[Feature] Support ThreeNN with cambricon MLU backend ( #2215 )
2022-08-23 15:18:47 +08:00
e663670a74
[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-21 23:21:54 +08:00
832b845257
[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-19 10:36:26 +08:00
b9bdfeb6f6
[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-17 10:53:05 +08:00
304f184c46
[Fix] Fix gradient cumulative optimizer when resuming ( #2093 )
...
* Fix gradient cumulative optimizer when resuming
* add unit test
2022-08-15 10:50:43 +08:00
b2ac245602
[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-08-03 16:21:41 +08:00
4558bfbd5b
[Fix] Fix onnx unit tests ( #2155 )
2022-07-29 19:23:29 +08:00
78f01001d5
[Feature] Support setting start epoch and interval epoch for PAVI Logger Hook ( #2103 )
...
* [Feature] Support setting start epoch and interval epoch for PAVI Logger Hook
* [Feature] Update the coding style as the maintainer wish
* fix: default integer division or modulo by zero
* fix: runner.epoch is less than start and use self.get_epoch instead of runner.epoch
* feat: support for iter-based runner and fix the step bug
* feat: iter based hook
* feat: fix bug and coding style
* fix: coding style
* fix: coding style
* fix: graph may add in evaluation
2022-07-29 11:40:20 +08:00
35affc35bf
[Feature] Support PrRoIPool operation ( #2113 )
...
* support prroipooling
* fix build bug
* fix bug of precision
* fix lint
* add ut
* add license and citation
* polish op and add type hint
* refactor ut
* add docstring for prroipool
* polish code
* format cu and cuh code
* fix format
* fix format
* fix typo
Co-authored-by: zhouzaida <zhouzaida@163.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-07-23 23:08:32 +08:00
22fadceecd
[Feature] Add MPS bbox overlap ( #2123 )
...
* add mps bbox overlap
* format
* update document and manifest
* update readme
2022-07-22 19:30:01 +08:00
6a03918f55
[Feature] Add support for mps ( #2092 )
...
* [Feature] Add support for MPS
* fix import error
* update ut
* fix error
* trigger CI
* use a unique basename for test file modules
* avoid bc-breaking
2022-07-07 16:05:49 +08:00
14c3e175fa
[Fix] num_channels can not be divisible by num_groups in test_build_norm_layer ( #2085 )
2022-06-30 19:28:34 +08:00
c03bba0008
Add missing json file for testing torchvision0.13 ( #2084 )
2022-06-30 19:12:11 +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
f5425ab761
Add torch_meshgrid wrapper due to PyTorch change ( #2044 )
...
* Add torch_meshgrid_ij wrapper due to PyTorch change
* Update torch_meshgrid name/doc/version implementation
* Make imports local
* add ut
* ignore ut when torch is not available
Co-authored-by: zhouzaida <zhouzaida@163.com>
2022-06-15 20:36:48 +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
d0dc7577b8
Transfer test data to aliyun ( #2004 )
2022-05-27 14:54:08 +08:00
05bdd1624f
Fix lint ( #1984 )
2022-05-25 20:45:31 +08:00
cb781ae4ba
[Fix] Fix the problem that the instance cannot reuse when overwriting the file client. ( #1747 )
...
* [Fix] Fix the problem that the instance cannot reuse when overwriting the file client.
* Remove _overridden_backends and _overridden_prefixes, and pop the old instance when overwriting the file client.
* add unittest
* fix prefix error
* modify clear_backend to overridden_backend
* Delete redundant comments.
2022-05-25 19:34:15 +08:00
c90f2be0be
[Fix] Fix is_module_wrapper ( #1900 )
...
* fix is_module_wrapper
* test is_module_wrapper
* fix code style
2022-05-25 17:32:41 +08:00
DaGaiBa
Zhongyu Li
jayggh
Zaida Zhou
mengpenghui
SUNYC
Shanghua Gao
ZShaopeng
ckirchhoff
q.yao
Zhang
VVsssssk
Yue Zhou
takuoko
cathyzhang222
Ivan Shcheklein
CokeDong
wangjiangben-hw
bdf
liuduanhui
Jingwei Zhang
Yosuke Shinya
Yixiao Fang
xcnick
Peter Ye
Mashiro
zhouchenyang
Philipp Allgeuer
Yanyi Liu
Danila Rukhovich
Wencheng Wu
whcao