453 Commits

Author SHA1 Message Date
q.yao
cb2eb57683
Remove onnx (#2295)
* remove onnx

* remove print
2022-10-21 15:35:07 +08:00
Mashiro
a6c42ad324
[Feature] Add TTA transform (#2146)
* Add TestRandomResize

* rename ut class

* minor retine

* Replace for loop with itertools.product

* Support accept built transforms

* Fix unit test

* Refine docstring

* minor refine

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

* Minor refine

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-10-19 23:32:22 +08:00
Miao Zheng
bf48ca035b
[Fix] Add swap_labe_pairs in RandomFlip (#2332)
* [Fix] Add `swap_labe_pairs` in `RandomFlip`

* [Fix] Add `swap_labe_pairs` in `RandomFlip`

* add reference info

* add swap_label_pairs in results

* revise according to comments

* revise according to comments

* revise according to comments

* docstring

* docstring
2022-10-17 23:21:20 +08:00
Yue Zhou
a4c82617ff
[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-13 17:26:14 +08:00
liuduanhui
cdfbdc0bd1 [Feature] Support ThreeNN with cambricon MLU backend (#2215) 2022-08-28 17:54:20 +08:00
bdf
625e82ce62 [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-28 17:54:20 +08:00
bdf
f5a19ef06f [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-28 17:54:20 +08:00
q.yao
2e5628b410
[Refactor]: Remove deployment for dev-2.x (#2225)
* remove deploy for 2.0

* update onnx ut
2022-08-26 20:11:05 +08:00
Zaida Zhou
c3835415ed
[Refactor] Modify the importing path due to mmengine changes (#2224) 2022-08-24 19:48:10 +08:00
Zaida Zhou
5947178e85
Remove many functions in utils and migrate them to mmengine (#2217)
* Remove runner, parallel, engine and device

* fix format

* remove outdated docs

* migrate many functions to mmengine

* remove sync_bn.py
2022-08-23 10:12:41 +08:00
Zaida Zhou
9185eee858
Remove runner, parallel, engine and device (#2216)
* Remove runner, parallel, engine and device

* fix format

* remove outdated docs
2022-08-22 19:16:30 +08:00
Mashiro
19a024155a
[Refactor] Use MODELS registry in mmengine and delete basemodule (#2172)
* change MODELS to mmengine, delete basemodule

* fix unit test

* remove build from cfg

* add comment and rename TARGET_MODELS to registry

* refine cnn docs

* remove unnecessary check

* refine as comment

* refine build_xxx_conv error message

* fix lint

* fix import registry from mmcv

* remove unused file
2022-08-19 11:18:47 +08:00
Jingwei Zhang
86f9dc7a40 [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-18 19:20:57 +08:00
Zaida Zhou
0fe1c64744
Remove fileio from mmcv and use mmengine.fileio instead (#2179) 2022-08-16 13:37:19 +08:00
Zaida Zhou
0b4285d959
Pick commits from master (#2164)
* [Docs] Add swith_language.md in docs (#2160)

* [Fix] Fix onnx unit tests (#2155)

* [Docs] Limit extension versions (#2144)

* Support PrRoIPool operation

* Add MPS bbox overlap

* Add .pre-commit-config-zh-cn.yaml (#2135)

Co-authored-by: xcnick <xcnick0412@gmail.com>
Co-authored-by: Jingwei Zhang <zjw18@mails.tsinghua.edu.cn>
Co-authored-by: q.yao <yaoqian@sensetime.com>
2022-08-03 14:11:53 +08:00
Yixiao Fang
e5c2cab583 [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-07-19 19:18:16 +08:00
plyfager
1a57a76be5 complete repr functions 2022-07-19 19:18:16 +08:00
liukuikun
f90567a091 [Fix] LoadImageFromFile 2022-07-19 19:18:16 +08:00
huanghaian
864942be2f fix resize error 2022-07-19 19:18:16 +08:00
Ma Zerun
a1fb002269 [Refactor] Use resize_type instead of resize_cfg in RandomResize and RandomChoiceResize. 2022-07-19 19:18:16 +08:00
YuanLiuuuuuu
a03774d6db [Refactor]: Grascale return uint8 type 2022-07-19 19:18:16 +08:00
YuanLiuuuuuu
5867a97a41 [Fix]: Fix random resize 2022-07-19 19:18:16 +08:00
liyining
59eaefebe0 [Feature] Support partial mapping by manually marking keys as ignored 2022-07-19 19:18:16 +08:00
gongtao.vendor
3b494a1304 Support broadcasting all keys for TransformBroadcaster 2022-07-19 19:18:16 +08:00
liukuikun
88f3cc3f35 replace height, width with img_shape 2022-07-19 19:18:16 +08:00
liukuikun
ab0aa4ef86 load type (#1921) 2022-07-19 19:18:16 +08:00
Yining Li
f59aec8ffb [Feature] Add decorator avoid_cache_randomness (#1864)
* add prohibit_cache_randomness

* rename as avoid_cache_randomness and ensure it is non-inheritable

* fix lint

* update docs
2022-07-19 19:18:16 +08:00
Yifei Yang
ea84b67449 update scales and img_shape (#1871) 2022-07-19 19:18:16 +08:00
Yuan Liu
30b3844621 [Fix]: Update img_shape in resize related transform (#1907) 2022-07-19 19:18:16 +08:00
Yining Li
ff0dfb74d7 add RandomApply (#1863) 2022-07-19 19:18:16 +08:00
Ma Zerun
428512f88a Rename LoadAnnotation to LoadAnnotations (#1850) 2022-07-19 19:18:16 +08:00
Yifei Yang
b7525fae1e [Fix] Fix typos, init funcs, cacheable method of part3 of data transforms (#1784)
* fix typos and move args into cfg

* update docstring

* fix as comment

* fix lint

* update transforms as discussed before

* linting

* fix as comment

* fix lint

* fix lint and also update according to PackInput

* remove precommit change

* rename cacheable method

Co-authored-by: liukuikun <641417025@qq.com>
2022-07-19 19:18:16 +08:00
liukuikun
e7592a70f2 [Fix] fix some typo in transforms (#1831)
* [Fix] fix some typo according to review

* fix comment
2022-07-19 19:18:16 +08:00
Yining Li
e2ca0733c6 Refactor base transform (#1830)
* rename cacheable_method to cache_randomness

* refactor transform wrappers and update docs

* fix all_nonexist_keys

* fix lint

* rename transform wrappers
2022-07-19 19:18:16 +08:00
Yuan Liu
0a5b412509 [Feature]: Reformat resize config (#1826)
* [Feature]: Add cache to random func in data transform

* [Fix]: Fix lint

* [Fix]: Fix cache decorate problem

* [Refactor]: Initialize Resize with config

* [Refactor]: Move other resize config into Resize config

* [Fix]: Scale can not be None in RandomResize

* [Fix]: Change semantic seg to gt seg map

* [Fix]: Delete unnecessary assert

* [Fix]: Fix docstring

* [Fix]: Add double quot to Resize in config

* [Fix]: Fix the return type

* [Fix]: Improve docstring

* [Fix]: Specify the order of width and height for ratio range

* [Fix]: Specify resize order

Co-authored-by: Your <you@example.com>
2022-07-19 19:18:16 +08:00
Yifei Yang
2844a11631 [Fix] Fix MultiScaleFlipAug (#1801)
* Fix MultiScaleFlipAug

* fix as comment
2022-07-19 19:18:16 +08:00
Yifei Yang
169f098d82 [Fix] Set keypoints not in the cropped image invisible (#1804)
* set invisiblity

* fix as comment
2022-07-19 19:18:16 +08:00
Yining Li
6534efd6ca [Fix] Add @cacheable_method in transform wrapper RandomChoice (#1807)
* add @cacheable_methd in RandomChoice

* RandomChoice add __iter__() and fix unittest
2022-07-19 19:18:16 +08:00
Yifei Yang
2f85d78149 [Enhancement] Enhance CenterCrop (#1765)
* enhance centercrop and adjust crop size to (w, h)

* fix comments

* update required keys and docstring
2022-07-19 19:18:16 +08:00
Yifei Yang
2619aa9c8e [Feature] Add Part3 of data transform (#1735)
* update data transform part3

* update init

* rename flip funcs

* fix comments

* update comments

* fix lint

* Update mmcv/transforms/processing.py

* fix docs format

* fix comments

* add test pad_val and fix bugs in class Pad

* merge updated pad

* fix lint

* Update tests/test_transforms/test_transforms_processing.py
2022-07-19 19:18:16 +08:00
liukuikun
5af6c12b81 [Fix] Pad boader fix (#1757)
* [Fix] expand pad_val'dim to be same with image

* fix comment
2022-07-19 19:18:16 +08:00
Yuan Liu
53070ebccf [Feature]: Add Part2 of data transform (#1730)
* [Refactor]: New commit of Part2 of data transform

* [Fix]: Fix lint

* [Fix]: Change flip reisze to prefix

* [Refactor]: Delete redundant code in ToTensor

* [Fix]:optional

* [Fix]: Change the discription of RandomFlip

* [Refactor]: Change flip_with_flip_direction to flip_on_direction

Co-authored-by: Your <you@example.com>
2022-07-19 19:18:16 +08:00
liukuikun
9e4b2ff58e [Feature] Add Part1 of data transform (#1736)
* [Feature] Add Part1 of data transform

* api.rst transform->transforms

* fix typo

* fix typo

* rename test_processing into test_transforms_processing for avoiding same name

* fix mypy

* fix comment

* fix comment
2022-07-19 19:18:15 +08:00
Yining Li
d00b0cec74 [Feature] Add base transform interface (#1538)
* Support deepcopy for Config (#1658)

* Support deepcopy for Config

* Iterate the `__dict__` of Config directly.

* Use __new__ to avoid unnecessary initialization.

* Improve according to comments

* [Feature] Add spconv ops from mmdet3d (#1581)

* add ops (spconv) of mmdet3d

* fix typo

* refactor code

* resolve comments in #1452

* fix compile error

* fix bugs

* fix bug

* transform from 'types.h' to 'extension.h'

* fix bug

* transform from 'types.h' to 'extension.h' in parrots

* add extension.h in pybind.cpp

* add unittest

* Recover code

* (1) Remove prettyprint.h
(2) Switch `T` to `scalar_t`
(3) Remove useless lines
(4) Refine example in docstring of sparse_modules.py

* (1) rename from `cu.h` to `cuh`
(2) remove useless files
(3) move cpu files to `pytorch/cpu`

* reorganize files

* Add docstring for sparse_functional.py

* use dispatcher

* remove template

* use dispatch in cuda ops

* resolve Segmentation fault

* remove useless files

* fix lint

* fix lint

* fix lint

* fix unittest in test_build_layers.py

* add tensorview into include_dirs when compiling

* recover all deleted files

* fix lint and comments

* recover setup.py

* replace tv::GPU as tv::TorchGPU & support device guard

* fix lint

Co-authored-by: hdc <hudingchang.vendor@sensetime.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>

* Imporve the docstring of imfrombytes and fix a deprecation-warning (#1731)

* [Refactor] Refactor the interface for RoIAlignRotated (#1662)

* fix interface for RoIAlignRotated

* Add a unit test for RoIAlignRotated

* Make a unit test for RoIAlignRotated concise

* fix interface for RoIAlignRotated

* Refactor ext_module.nms_rotated

* Lint cpp files

* add transforms

* add invoking time check for cacheable methods

* fix lint

* add unittest

* fix bug in non-strict input mapping

* fix ci

* fix ci

* fix compatibility with python<3.9

* fix typing compatibility

* fix import

* fix typing

* add alternative for nullcontext

* fix import

* fix import

* add docstrings

* add docstrings

* fix callable check

* resolve comments

* fix lint

* enrich unittest cases

* fix lint

* fix unittest

Co-authored-by: Ma Zerun <mzr1996@163.com>
Co-authored-by: Wenhao Wu <79644370+wHao-Wu@users.noreply.github.com>
Co-authored-by: hdc <hudingchang.vendor@sensetime.com>
Co-authored-by: grimoire <yaoqian@sensetime.com>
Co-authored-by: Jiazhen Wang <47851024+teamwong111@users.noreply.github.com>
Co-authored-by: Hakjin Lee <nijkah@gmail.com>
2022-07-19 19:18:15 +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
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
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
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