1233 Commits

Author SHA1 Message Date
tripleMu
966b742817
Add type hints for mmcv/runner/optimizer (#2001) 2022-05-28 23:45:42 +08:00
ShawnHu
1577f40744
Add type hints in mmcv/ops/active_rotated_filter.py (#2017)
* add typehints in ops-active-rotated-filter

* resolve typehints in ops-active-rotated-filter
2022-05-28 19:57:27 +08:00
gy77
19902d897a
Add type hints in mmcv/ops/points_sampler.py (#2015)
* add typehint in mmcv/ops/points_sampler.py

* Update mmcv/ops/points_sampler.py

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

* Update mmcv/ops/points_sampler.py

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

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-28 15:35:17 +08:00
gy77
1211b06b9e
Add type hint in mmcv/ops/points_in_boxes.py (#2007)
* add type hint in mmcv/ops/points_in_boxes.py

* fix lint
2022-05-28 11:57:34 +08:00
gy77
50d1505225
Add type hint in mmcv/ops/points_in_polygons.py (#2013) 2022-05-28 11:57:08 +08:00
WINDSKY45
335199db92
Add type hint in lr_updater.py (#1988)
* [Enhance] Add type hint in `lr_updater.py`.

* Fix circle import
2022-05-27 15:43:47 +08:00
gy77
23eb359beb
Add type hint for the "boxes_iou_bev" function in iou3d.py (#2005)
* add type hints in iou3d.py

* fix lint

* Add a missing item in iou3d.py
2022-05-27 15:01:01 +08:00
Zaida Zhou
d0dc7577b8
Transfer test data to aliyun (#2004) 2022-05-27 14:54:08 +08:00
kira
a465fb4114
Add type hints in alexnet.py (#1983)
* Update alexnet.py

* Update alexnet.py

* Update mmcv/cnn/alexnet.py

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

* Update alexnet.py

* Update mmcv/cnn/alexnet.py

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

* Update mmcv/cnn/alexnet.py

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

* Update alexnet.py

* fix importing format

* Update alexnet.py

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-27 14:00:18 +08:00
zengxiang68
699398ad86
Add type hints for mmcv/ops (#1987)
* add type hints for mmcv/ops/...

* add type hints for mmcv/ops/...

* add type hints for mmcv/ops/...
2022-05-27 12:45:47 +08:00
Yvette Zhao
84a544fb3e
Add type hints for runner/builder (#1985) 2022-05-27 10:44:08 +08:00
tripleMu
ef2b927b29
Add typehint in mmcv/utils* (#1998)
* Add typehint in mmcv/utils*

* Update mmcv/utils/registry.py

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

* Update mmcv/utils/registry.py

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

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-27 10:30:25 +08:00
gy77
af861bf9b8
Add type hints in iou3d.py (#1989)
* add type hints in iou3d.py

* fix lint
2022-05-27 10:23:10 +08:00
Range King
bb4c65d8ff
Add type hints in bbox.py and box_iou_rotated.py (#1981)
* Add type hints in bbox.py

* Add type hints in box_iou_rotated.py

* Update ball_query.py

* Revert "Add type hints in bbox.py"

This reverts commit 102566488e3aa9328b1c6e1159e2699d40f9f9f5.

* Update bbox.py

* Revert "Update bbox.py"

This reverts commit d34b2c6ed54ba024b9674c948493001a8b4ff864.

* Update bbox.py
2022-05-27 10:21:48 +08:00
Zaida Zhou
1e922b84ee
[Fix] Fix protobuf compatible error (#1992) 2022-05-26 17:26:45 +08:00
Zaida Zhou
05bdd1624f
Fix lint (#1984) v1.5.2 2022-05-25 20:45:31 +08:00
Zaida Zhou
4ea914ed28
bump version to v1.5.2 (#1979) 2022-05-25 20:08:02 +08:00
Wencheng Wu
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
wlf-darkmatter
544a6b1432
Imporve ability to view usage data during training (#1940)
* Imporve ability to view usage data during training

The wrong labeling of the dataset may cause problems such as gradient explosion during the training process. The wrong labeling can be found by setting up a hook function.
**Problem**: However, when a for loop is used to traverse the iterator of the pytorch `DataLoader` class, the hook cannot obtain the information of the currently read image, and thus cannot determine the source of the error.
**Solution**: Load the data_batch information into the `runner` during the train process, and then pass it to the hook function to solve the problem.

* Update epoch_based_runner.py

* Update iter_based_runner.py

* strict the scope of runner.data_batch

* strict the scope of runner.data_batch

* strict the scope of runner.data_batch
2022-05-25 19:33:27 +08:00
Alex Yang
21bada3256
[fix]:fix docstrings in EvalHook (#1978) 2022-05-25 19:14:37 +08:00
whcao
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
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
del-zhenwu
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
JarvisKevin
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
Zaida Zhou
45fa3e44a2
Add pyupgrade pre-commit hook (#1937)
* add pyupgrade

* add options for pyupgrade

* minor refinement
2022-05-18 11:47:14 +08:00
Zaida Zhou
c561264d55
[Docs] Fix failed links in docs/zh_cn/faq.md (#1964)
* [Docs] Fix failed links in docs/zh_cn/faq.md

* fix links
2022-05-17 10:45:19 +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
q.yao
8708851eca
[Fix] Fix bbox overlap fp16 (#1958)
* add CUDA_ARCH check

* add check in cuh
v1.5.1
2022-05-14 19:46:27 +08:00
Zaida Zhou
a3b4640be8
bump version to v1.5.1 (#1951) 2022-05-13 10:50:11 +08:00
SuperDong
940609d5ea
[Enhancement] Add type annoations for mmcv/visualization/image (#1946)
* dev_1st_mmcv_visual_image_py

* dev_3rd_mmcv_visual_image_py

* dev_4th_mmcv_visual_image_py

* dev_5th_mmcv_visual_image_py

* dev_6th_mmcv_visual_image_py

* dev_7th_mmcv_visual_image_py

* dev_8th_mmcv_visual_image_py

* dev_9th_mmcv_visual_image_py

* minor fix

* fix isort

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-12 12:50:29 +08:00
Song Lin
94cc99d595
[Enhancement] Add type hints for mmcv/arraymisc and mmcv/video (#1950)
* Add type hints

* Add type hints

* Fix int float about scalar

* Add type hints for mmcv/tensorrt

* Update mmcv/tensorrt/tensorrt_utils.py

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

* Update mmcv/arraymisc/quantization.py

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

* Ignore type hint for dtype

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-11 21:55:41 +08:00
Song Lin
882cab77bb
[Enhancement] Add type annotations for mmcv/visualization (#1947)
* Add type hints

* Replace mmcv is_str by isinstance str

* Add type hints in optflow.py

* Update mmcv/visualization/optflow.py

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

* Update mmcv/visualization/color.py

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

* Fix tuple hints

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-10 23:48:00 +08:00
Jiazhen Wang
105fe55002
[Feature] Add adjust_hue (#1628)
* [Impl] impl adjust hue

* Update docstring

* Fix docstring

* add copyright

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-10 17:06:45 +08:00
jmercat
1dbb5d360d
[Fix] Fix dump method of Config (#1837)
* config dump fix
dump should not depend on how config was loaded only where it is dumped
changed checks for type to the given filename instead of the source filename

* defined test_fromdict and fixed dump function
it might be more convoluted than it needs to be but the formatting depends on the extension of the argument or the extension of the self.filename

* config dump defaults to returning pretty_text, test_fromdict renamed to test_dump_from_dict

* some reformatting in docstrings

* refine unittest

* fix unit test as comment

* Refine docstring

* minor refinement

* import mmcv

* fix lint

Co-authored-by: HAOCHENYE <21724054@zju.edu.cn>
Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-10 15:41:51 +08:00
Alex Yang
a848ecfdfc
[Fix] Fix type hint in file_client (#1942)
* [fix]:fix type hint in file_client and mmcv

* [fix]:fix type hint in tests files

* [fix]:fix type hint in tests files

* [fix]:fix pre-commit.yaml to igore test for mypy

* [fix]:fix pre-commit.yaml to igore test for mypy

* [fix]:fix precommit.yml

* [fix]:fix precommit.yml

* Update __init__.py

delete unused type-ignore comment
2022-05-10 14:01:07 +08:00
Zaida Zhou
fb9af9f36f
[Docs] Fix heading anchor in documentation (#1945) 2022-05-09 22:53:17 +08:00
Alex Yang
19b1d363a5
[Fix] Incorrectly append items to __all__ (#1944) 2022-05-09 19:45:45 +08:00
pc
e70a318ae9
[Fix] Fix parrots unit tests (#1913) 2022-05-09 16:59:32 +08:00
Hongbin Sun
a86e807105
[Fix] LmdbBackend can't pickle env obj (#1764)
* fix bug of can't pickle env objs

* use utf-8 to support Chinese

* use __del__ instead close

* move try import to init

* remove getitem

* use client instead
2022-05-08 17:21:52 +08:00
Johannes L
d126f3026c
[Enhancement] Add support for latest Pillow resampling filter naming scheme (#1931)
* Add support for latest Pillow resampling filter naming scheme

* Comments added

* Shortened comments, lint fixes

* Added specific Pillow version
2022-05-08 17:20:49 +08:00
Zaida Zhou
8b4dcf1dfb
[Docs] Include several files in index.rts (#1939) 2022-05-07 16:26:18 +08:00
Boyuan Deng
8ecaf2e8b2
[Enhancement] Set environment variables in _init_dist_mpi (#1682)
* [Enhancement] Initialize 'mpi' launcher from OpenMPI's environmental variables

* raise exception when no MASTER_ADDR

Co-authored-by: jzwang <841713301@qq.com>
2022-05-05 12:36:52 +08:00
Mashiro
887d9a0f38
[Feature] Add a script to visualize lr (#1667)
* add learning rate visualization

* support momentum visualization

* fix second axis font size error

* fix argparse description

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

* fix lint

* Accelerate visualization

* Support custom batchsize and log-interval

Support custom batchsize and log-interval

* remove unnecessary keys and rename some varaible

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-05 12:29:03 +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
pc
9f5a03dc2c
[Feature] Add diff_iiou_rotated op in parrots (#1911) 2022-04-30 09:43:33 +08:00
Zaida Zhou
057c032347
[Docs] Update wechat in README_zh-CN.md (#1927) 2022-04-29 14:11:38 +08:00
Zaida Zhou
0cc84451df
[Fix] Fix GPG key error in CI (#1926) 2022-04-29 13:17:31 +08:00
WilliamKyle
b30755eef2
[Feature] Add rotated_feature_align cpu & onnxruntime implementation (#1878)
* add rotated_feature_align cpu implementation

* add rotated_feature_align onnxruntime implementation

* Update code for advices from grimoire

Remove useless comment from mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp

Replace ambiguous function name atomicAdd in mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp

Simplify unit test with parameter in tests/test_ops/test_rotated_feature_align.py

Use fma in interpolate in mmcv/ops/csrc/onnxruntime/cpu/rotated_feature_align.cpp mmcv/ops/csrc/pytorch/cpu/rotated_feature_align.cpp

* Inline the function to reduce the overhead of the function call

Use fma in interpolate
2022-04-29 10:02:33 +08:00
Hu Di
4efec73269
[Docs] Add IPU installation documentation (#1904)
* add doc for mmcv installation

* Update docs/en/get_started/build.md

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

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-04-28 14:59:27 +08:00
q.yao
30057a6113
[Fix] Fix nms rotate illegal memory access (#1891) 2022-04-27 21:06:30 +08:00