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
b1711db048
[Refactor] Adapt mlu code to cntoolkit3.0.1 ( #2248 )
...
* [Refactor] Adapt mlu code to cntoolkit3.0.1
* [Refactor] Modify the code style
Co-authored-by: budefei <budefei@cambricon.com>
Co-authored-by: budefei <budefei@cambricon>
2022-10-10 22:40:41 +08:00
8e844d1b5c
Fix warning of CUDA ops ( #2324 )
2022-10-10 14:07:40 +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
92504176bc
Update onnx registry ( #2294 )
...
* update onnx registry
* move import
2022-09-29 22:12:08 +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
847612dfff
Bump version to v1.6.2 ( #2285 )
2022-09-26 19:02:26 +08:00
629ef89c0d
Pick commits from 2.x branch ( #2284 )
...
* [Fix] Fix the calculation error of out_w in MaskedConv2dFunction (#2264 )
* [Fix] Fix two intergers can not use / operator in torch1.6
* should not use kernel_h when calculates out_w
* [Docs] Fix some mistakes in data_process.md (#2283 )
* correction
* correction en docs
Co-authored-by: yxzhao <33384732+yxzhao2022@users.noreply.github.com>
2022-09-25 23:26:04 +08:00
3b9b9054bf
[Enhancement] Add support of TorchVision's Model Registration API on MMCV ( #2246 )
...
* Add support of TorchVision's Model Registration API.
* fix linter
* formatting with yapf
2022-09-14 16:41:35 +08:00
25f533b916
Fix spconv clang-format ( #2260 )
2022-09-14 11:01:09 +08:00
55c51e1734
[Feature] Register a new activatation layer SiLU to ACTIVATION_LAYERS ( #2256 )
2022-09-13 22:38:23 +08:00
2046a394a2
[Fix] Use ROCm backened within the PyTorch framework ( #1918 )
...
* modified code for using ROCm backened within the PyTorch framework
* added hip runtime header
* flake8 linting fix
2022-09-13 15:43:48 +08:00
2fb2b91aed
Add type hints for mmcv/image ( #2089 )
...
* Fix typehint
* minor fix
* minor fix
* minor fix
Co-authored-by: zhouzaida <zhouzaida@163.com>
2022-09-12 17:16:55 +08:00
3a311e85ae
[Docs] Add zhihu articles ( #2157 )
...
* [Docs] Add zhihu articles
* add OpenMMLabCourse link
* fix typo
2022-09-11 22:21:52 +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
e08b80911f
No restritions on onnx versions ( #2251 )
2022-09-08 15:44:27 +08:00
7f09a983c3
[Docs] Fix the example error in registry.md ( #2243 )
...
the lack of colon(:)
2022-09-02 11:37:45 +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
505b5771eb
[Fix] cast the type of mask to enable training with amp ( #2220 )
2022-08-28 22:26:58 +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
9b11e560f3
[Fix] Fix roi_align_rotated op of MLU ( #2210 )
...
* [Fix] roi_align_rotated codes
* [Fix]: fix code style
* [Fix]: fix code style
* [Fix]: fix code style
2022-08-19 21:19:40 +08:00
730a53a062
[Fix] Path mapping only once in storage backends ( #2204 )
...
* [Fix] Path mapping only once in storage backends
* Fix lint
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-08-19 10:38:27 +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
90e19c1c6e
[Fix] Fix a potential bug in prroipool op ( #2200 )
2022-08-18 15:02:20 +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
de9e625f4b
[Feature] Add argument params to MlflowLoggerHook ( #2193 )
2022-08-16 11:54:21 +08:00
595c2ebeb9
[Feature] Add ChamferDistance op for Parrots ( #2189 )
...
* Support Parrots extension for op ChamferDistance
* Fix lint
* Adapt op backward error of IntTensor for parrots
* Fix Lint
2022-08-16 00:37:57 +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
f527e43c1a
[Fix] Fix init weights of MultiScaleDeformableAttention ( #2158 )
...
* fix tensors on different device
* fix lint
* Update mmcv/ops/multi_scale_deform_attn.py
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-08-03 16:21:41 +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
bad822d784
[Docs] Add swith_language.md in docs ( #2160 )
2022-07-30 12:43:15 +08:00
4558bfbd5b
[Fix] Fix onnx unit tests ( #2155 )
2022-07-29 19:23:29 +08:00
1248da3cea
[Fix] Fix the upload graph error of PaviLoggerHook. ( #2100 )
...
* [Fix] Fix add_graph function of PaviLoggerHook.
* Fix circular reference error.
* Add comments.
* modify partial_args default to empty dict.
* Add warnning for add_graph and img_key parameter.
2022-07-29 17:24:35 +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
eb4bbbbd64
[Docs] Limit extension versions ( #2144 )
2022-07-25 14:12:27 +08:00
d409eedc81
bump version to v1.6.1 ( #2142 )
2022-07-23 23:39:33 +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
73066430be
Add .pre-commit-config-zh-cn.yaml ( #2135 )
2022-07-22 10:31:24 +08:00
d1004b7c92
Update daily dispatcher ( #2125 )
2022-07-19 19:12:54 +08:00
0092699fef
[Enhancement] Improve ProfilerHook beaviour when using a relative (or empty) dir_name ( #2119 )
...
* Add default dir_name and make non-absolute dir_name relative to runner.work_dir
* Add log message and change default dir name
2022-07-18 19:40:59 +08:00
b58c84c6f0
[Docs] Fix markdown version ( #2121 )
...
* [Docs] Fix sphinx markdown tables version
* fix markdown version
2022-07-17 22:03:28 +08:00
323abb9e5a
[Docs] Fix typos ( #2124 )
2022-07-17 18:41:17 +08:00
75ae200954
[Docs] Correct mistakes of examples ( #2120 )
2022-07-16 11:33:43 +08:00
f4167fe1e3
Bump version to v1.6.0 ( #2096 )
...
* bump version to v1.6.0
* minor fix
2022-07-08 20:32:52 +08:00
da2df84a9f
[Fix] Make `MMDistributedDataParallel` compatible with Pytorch1.12 ( #2107 )
...
* make compatible with pytorch 1.12
* override _run_ddp_forward
* over write _run_ddp_forward
* refactor docstring
2022-07-08 16:14:35 +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
CokeDong
q.yao
bdf
tpoisonooo
Zaida Zhou
wangjiangben-hw
takuoko
Vasilis Vryniotis
Zachary Streeter
tripleMu
ZShaopeng
Hoantrbl
Yanhong Zeng
liuduanhui
tudejiang79
Miao Zheng
Jingwei Zhang
eivtho
Yosuke Shinya
Cedric Luo
Yixiao Fang
xcnick
Wencheng Wu
Peter Ye
GPhilo
Reza
Happylkx
Mashiro