Commit Graph

1217 Commits (2fb2b91aedc0955b6cab3858d9d2201f4b1bf328)
 

Author SHA1 Message Date
tripleMu 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
Zaida Zhou 3a311e85ae
[Docs] Add zhihu articles (#2157)
* [Docs] Add zhihu articles

* add OpenMMLabCourse link

* fix typo
2022-09-11 22:21:52 +08:00
ZShaopeng 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
Zaida Zhou e08b80911f
No restritions on onnx versions (#2251) 2022-09-08 15:44:27 +08:00
Hoantrbl 7f09a983c3
[Docs] Fix the example error in registry.md (#2243)
the lack of colon(:)
2022-09-02 11:37:45 +08:00
bdf 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
Yanhong Zeng 505b5771eb [Fix] cast the type of mask to enable training with amp (#2220) 2022-08-28 22:26:58 +08:00
liuduanhui a08517790d
[Feature] Support ThreeNN with cambricon MLU backend (#2215) 2022-08-23 15:18:47 +08:00
bdf 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
tudejiang79 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
Miao Zheng 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
bdf 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
Jingwei Zhang 90e19c1c6e
[Fix] Fix a potential bug in prroipool op (#2200) 2022-08-18 15:02:20 +08:00
Jingwei Zhang 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
eivtho de9e625f4b
[Feature] Add argument params to MlflowLoggerHook (#2193) 2022-08-16 11:54:21 +08:00
CokeDong 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
Yosuke Shinya 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
Cedric Luo 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
Yixiao Fang 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
Zaida Zhou bad822d784
[Docs] Add swith_language.md in docs (#2160) 2022-07-30 12:43:15 +08:00
xcnick 4558bfbd5b
[Fix] Fix onnx unit tests (#2155) 2022-07-29 19:23:29 +08:00
Wencheng Wu 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
Peter Ye 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
Zaida Zhou eb4bbbbd64
[Docs] Limit extension versions (#2144) 2022-07-25 14:12:27 +08:00
Zaida Zhou d409eedc81
bump version to v1.6.1 (#2142) 2022-07-23 23:39:33 +08:00
Jingwei Zhang 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
q.yao 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
Zaida Zhou 73066430be
Add .pre-commit-config-zh-cn.yaml (#2135) 2022-07-22 10:31:24 +08:00
Zaida Zhou d1004b7c92
Update daily dispatcher (#2125) 2022-07-19 19:12:54 +08:00
GPhilo 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
Zaida Zhou b58c84c6f0
[Docs] Fix markdown version (#2121)
* [Docs] Fix sphinx markdown tables version

* fix markdown version
2022-07-17 22:03:28 +08:00
Reza 323abb9e5a
[Docs] Fix typos (#2124) 2022-07-17 18:41:17 +08:00
Happylkx 75ae200954
[Docs] Correct mistakes of examples (#2120) 2022-07-16 11:33:43 +08:00
Zaida Zhou f4167fe1e3
Bump version to v1.6.0 (#2096)
* bump version to v1.6.0

* minor fix
2022-07-08 20:32:52 +08:00
Mashiro 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
Zaida Zhou 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
Yuchen Fan 357b484d37
[Fix] Fix undefined variable error in pixel_group.cpp (#2104) 2022-07-07 15:32:25 +08:00
Grapymage d77557b08c
[Fix] Fix index error when using multi-samplers strategy (#2094)
* fix index error when using multi-samplers strategy

* After every loop, change the last_fps_end_index to fps_sample_range
2022-07-06 20:44:19 +08:00
Evelyn Wang c67ab9a356
[Enhancement] Refine import error of PAVI (#2098)
* Import Error message refined.

* PAVI import error message refined.
2022-07-05 14:23:19 +08:00
Zaida Zhou 9fffcbc9cd
[Feature] Add Dockerfile (#1974)
* Add Dockerfiles

remove Dockerfile in root directory

minor fix

minor fix

* improve dockerfile

* refine
2022-07-01 16:23:28 +08:00
Zaida Zhou c344c11004
Test torch1.12 in CI (#2088) 2022-06-30 21:57:09 +08:00
Mashiro 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
Mashiro c03bba0008
Add missing json file for testing torchvision0.13 (#2084) 2022-06-30 19:12:11 +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
q.yao d71d067da1
[Fix] Fix mdcn trt (#2075) 2022-06-26 21:54:49 +08:00
Evelyn Wang 5e6ede9946
[Enhancement] Improve the docstring of PaviLoggerHook (#2060)
* Add parameter configuration description of summarywriter to MMCV PAVI hook.

* Typo repaired

* fix typo with suggested change

* self.init_kwargs lines modified.
2022-06-26 21:25:43 +08:00
tripleMu 1f25001028
Add type hints for mmcv/runner (#2002)
* Fix

* Fix

* fix type hint

* minor fix

* remove some type hints of functions or methods

* minor fix

* Apply suggestions from code review

* minor fix

* minor refinement

Co-authored-by: HAOCHENYE <21724054@zju.edu.cn>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: zhouzaida <zhouzaida@163.com>
2022-06-20 18:51:40 +08:00
WINDSKY45 b9a96e5677
[Enhance] Add type hints in /ops: (#2030)
* [Enhance] Add type hints in /ops:
`fused_bias_leakyrelu.py`, 'gather_points.py`, `group_points.py`.
There is no need to add type hints in `furthest_point_sample.py` and
`info.py`.
As for `focal_loss.py`, please see #1994.

* Modidied the default value of a variable.

* [Enhance] Add type hints in:
`knn.py`, `masked_conv.py`, `merge_cells.py`, `min_area_polygons.py`,
`modulated_deform_conv.py`, multi_scale_deform_attn.py`.

* Fix typehint.

* Fixed typehint.

* remove type hints of symbolic

* add no_type_check to ignore mypy check for method

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-06-20 14:53:23 +08:00
tripleMu 305c2a3025
Add type hints for mmcv/cnn/bricks (#1993)
* Add type hint

* Add typehint in mmcv/cnn/bricks*

* Deal conflict0

* Fix

* fix

* minor fix

* minor fix

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-06-20 14:52:45 +08:00
tripleMu 2d3e42fc41
Add type hints for mmcv/ops (#1995)
* Merge Master

* Add typehint in mmcv/ops/*

* Fix

* Update mmcv/ops/roi_align.py

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>

* Fix

* Fix

* Fix

* Update mmcv/ops/riroi_align_rotated.py

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

* Update mmcv/ops/riroi_align_rotated.py

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

* remove type hints of all symbolic methods

* remove type hints of all symbolic methods

* minor refinement

* minor refinement

* minor fix

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: zhouzaida <zhouzaida@163.com>
2022-06-20 14:50:28 +08:00