Commit Graph

1331 Commits (master)
 

Author SHA1 Message Date
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
Yvette Zhao 3dd2a21b45
Add type hints for runner/base_runner (#2003)
* Add type hints for runner

* refine

* fix error

* refine

* refine format

Co-authored-by: zhouzaida <zhouzaida@163.com>
2022-06-20 14:45:36 +08:00
Tri Wahyu Utomo bb7106755e
Add type hints for mmcv/runner (#1991)
* add typehints

* fix typehints

* fix _init_rule type hint

* update error message

* simplify type hint

* minor refinement

* minor refinement

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: HAOCHENYE <21724054@zju.edu.cn>
2022-06-15 22:41:10 +08:00
Philipp Allgeuer 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
tripleMu b062468015
Add type hints for mmcv/parallel (#2031)
* Add typehints

* Fix

* Fix

* Update mmcv/parallel/distributed_deprecated.py

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

* Fix

* add type hints to scatter

add type hints to scatter

* fix ScatterInputs

* Update mmcv/parallel/_functions.py

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

* Fix

* refine type hints

* minor fix

Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
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-15 20:28:15 +08:00
Alex Yang 9110df94fc
[Docs] Reformat op docs (#2052)
* [Docs]:update ops.md

* fix missing ops
2022-06-15 10:57:29 +08:00
Zaida Zhou 30f3058107
Bump version to v1.5.3 (#2053) 2022-06-11 18:48:50 +08:00
pc 9f30496cb0
[Fix] Fix iou3d in parrots (#2054) 2022-06-11 01:09:28 +08:00
Yanyi Liu 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
nxznm 5a2906cbd7
Add type hints in vgg.py (#2050)
* add type hints in vgg.py

* Update mmcv/cnn/vgg.py

* add type hints for return value in vgg.py
2022-06-09 22:26:32 +08:00
Danila Rukhovich 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
Zaida Zhou 11d8554cf9
[Docs] Fix mdformat version to support python3.6 (#2014)
* [Docs] Refine documentation

* fix mdformat version to support python3.6

* remove ruby dependency from CI

* fix invalid links
2022-06-09 17:47:26 +08:00
Yue Zhou 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
zhouchenyang 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
zhouchenyang 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
tripleMu 27d1d7fef2
Add type hints for mmcv/cnn/utils (#1996)
* Add typehint in mmcv/cnn/utils/*

* Update mmcv/cnn/utils/flops_counter.py

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

* Update mmcv/cnn/utils/weight_init.py

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

* Fix bugs

* Fix

* Fix2

* Update mmcv/cnn/utils/weight_init.py

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

* Update mmcv/cnn/utils/flops_counter.py

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

* Update mmcv/cnn/utils/weight_init.py

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

* Update mmcv/cnn/utils/weight_init.py

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

* Fix

* minor fix

* line is too long.

* Update mmcv/cnn/utils/weight_init.py

Co-authored-by: Jiazhen Wang <47851024+teamwong111@users.noreply.github.com>

* Fix

* Update mmcv/cnn/utils/weight_init.py

* fix default value of float type hint

* fix default value of float type hint

* fix default value of float type hint

* fix default value of float type hint

* fix default value of float type hint

* fix default value of float type hint

* fix default value of float type hint

* Fix

* minor refinement

* replace list with tuple for input

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Jiazhen Wang <47851024+teamwong111@users.noreply.github.com>
Co-authored-by: Mashiro <57566630+HAOCHENYE@users.noreply.github.com>
Co-authored-by: zhouzaida <zhouzaida@163.com>
2022-06-07 21:00:29 +08:00
pc 0230fc3bc0
[Feature] Add rotated_feature_align_cpu and update iod3d in parrots (#2027) 2022-06-07 16:20:06 +08:00
tc_exe 4061fcdc95
Add type hints for mmcv/engine/test.py (#2034)
* Add type hints for mmcv/engine/test.py

* Add type hints for mmcv/engine/test.py

* Add type hints for mmcv/engine/test.py

* fix type hint

Co-authored-by: zhouzaida <zhouzaida@163.com>
2022-06-07 16:12:07 +08:00
Haobo Yuan 3ede2789cc
[Fix] Include torchvision_12.json in the sdist package (#2039) 2022-06-07 10:16:08 +08:00
古月闻星 70944a261d
[Docs] Update copyright years in docs (#2042) 2022-06-06 23:20:49 +08:00
ShawnHu dcdf89611e
Add type hints for mmcv/ops (#2037)
* Add type hints for mmcv/ops/ball_query.py, border_align.py and correlation.py

* Add type hints for mmcv/ops/deform_conv.py, deform_roi_pool.py and deprecated_wrappers.py

* Remove type hints for deform_conv.py and deform_roi_pool.py

* Fix type hints for other files
2022-06-06 23:10:06 +08:00
WINDSKY45 ea0e8cdb41
[Enhancement] Add type ints in several files (#2020)
* [Enhance] Add type ints in these files:
'base_module.py', 'checkpoint.py',
'default_constructor.py',
'dist_utils.py', 'fp16_utils.py',
'log_buffer.py', 'priority.py'.

* Fixed all the inappropriate type hints.
Removed the return type of __init__ funcs.

* Fixed type hint.

* Added type hints in `dist_utils.py` and `log_buffer.py`.

* Remove unnecessary import.

* minor fix

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-06-03 07:19:12 +08:00
tripleMu 88ae2a4e59
Add type hints in several files (#1999)
* Add typehint in such files

* Update mmcv/video/processing.py

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

* Fix np.ndarray str

* Fix

* Update mmcv/video/processing.py

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

* Fix

* Fix

* Update mmcv/video/processing.py

Co-authored-by: Jiazhen Wang <47851024+teamwong111@users.noreply.github.com>

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: Jiazhen Wang <47851024+teamwong111@users.noreply.github.com>
2022-06-03 07:14:51 +08:00
WINDSKY45 85c0f85bea
[Enhancement] Add type hints in mmcv/ops/focal_loss.py. (#1994)
* [Enhance] Add type hint in `focal_loss.py`.

* Use torch.LongTensor as type of target.

* Fixed the missing type hint.

* Removed some unnecessary type hint.

* Add missing type hint and fix the type hint of `target`.

* Fix the format of typehint.

* minor refinement

* minor fix

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-06-02 23:32:01 +08:00
ShawnHu f51bcf50ed
Add type hints in mmcv/cnn/resnet.py (#1982)
* Add type hints in resnet.py

* using lint while commit

* using lint while commit

* using lint while commit

* reslove typehints

* add pre-defined type hints

* Add type hints for other methods in mmcv/cnn/resnet.py

* Fix type hints
2022-06-02 19:17:19 +08:00
ShawnHu aea2bb28d4
Add type hints for mmcv/ops (#2032)
* Add type hints in mmcv/ops/carafe.py

* Add type hints in mmcv/ops/corner_pool.py

* Add type hints in mmcv/ops/diff_iou_rotated.py

* Add type hints for other methods for mmcv/ops/corner_pool.py

* Add type hints for other methods in mmcv/ops/carafe.py

* Add type hints for symbolic method
2022-06-01 21:02:03 +08:00
WINDSKY45 bcd3291431
Add type hints for mmcv/runner (#2008)
* [Enhance] Add type hint in 'utils.py'.

* [Enhance] Add type hint in 'default_constructor.py'.

* Fixed undefined name.

* minor refinement

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-06-01 12:35:09 +08:00
gy77 d499abf2fd
Add type hints in mmcv/ops/pixel_group.py (#2016)
* add typehint in mmcv/ops/pixel_group.py

* fix indent

* fix indent

* improve format

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-31 10:06:05 +08:00
Zaida Zhou 9ea55badcb
[Fix] dump has incompatible type TextIO (#2029) 2022-05-31 00:02:37 +08:00
tripleMu 569d588e15
Add type hints for mmcv/fileio (#1997)
* Add typehint in mmcv/fileio/*

* Update mmcv/fileio/io.py

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

* Update mmcv/fileio/io.py

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

* Update mmcv/fileio/io.py

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

* Update mmcv/fileio/io.py

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

* Update mmcv/fileio/io.py

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

* Update mmcv/fileio/io.py

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

* Update mmcv/fileio/parse.py

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

* Update mmcv/fileio/parse.py

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

* Update mmcv/fileio/parse.py

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

* Update mmcv/fileio/io.py

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

* Update mmcv/fileio/io.py

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

* Fix hint bugs

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-05-30 17:42:53 +08:00
ShawnHu e234c183cc
Add type hints in ops/border_align.py (#2026) 2022-05-30 15:33:18 +08:00
tripleMu ea173c9f07
Add type hints for mmcv/runer/hooks/logger (#2000)
* Fix

* Fix
2022-05-30 15:31:15 +08:00
ShawnHu c70fafebf4
Add type hints in ops/assign_score_withk.py (#2023) 2022-05-29 20:55:50 +08:00
kira de90c7a2fd
[Enhancement] Update the missing warning message for LrUpdaterHook (#2021) 2022-05-29 11:31:47 +08:00
gy77 e5d9ef5285
Add type hints in mmcv/ops/point_sample.py (#2019)
* Add type hints in mmcv/ops/point_sample.py

* fix docstring and type hint
2022-05-29 09:25:27 +08:00
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