Commit Graph

1342 Commits (1db3967e860569c1501af4bff6c0964dec402289)
 

Author SHA1 Message Date
Zaida Zhou 1db3967e86
Fix lint (#2702) 2023-03-26 02:06:28 +08:00
Mashiro 5b0cdccc5c
[Fix] Enhance the compatibility of training stylegan 2 (#2694)
* [Fix] Enhance the compatibility of training stylegan 2

* Fix unit test in PyTorch 2.0

* Apply suggestions from code review

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-26 01:58:38 +08:00
q.yao aa250d0afa
[Fix] Replace torch/extension.h with torch/types.h to fix compilation error on Windows (#2698)
* replace extension to types

* update setup

* Update setup.py

* Update setup.py

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-25 11:42:08 +08:00
q.yao a6d7dde77f
[Fix] Fix torch2.0 dcn/mdcn symbolic (#2695)
* fix

* fix lint
2023-03-24 13:01:56 +08:00
Zaida Zhou 5a45fac989
Do not build pre-built pakcages for torch1.6 and 1.7 on macOS platform (#2678)
* build macos

* fix python version

* fix python version

* Do not build pre-built pakcages for torch1.6 and 1.7 in macOS platform
2023-03-21 13:54:00 +08:00
Zaida Zhou 6a5cd17702
Bump version to v2.0.0 (#2675) 2023-03-21 10:23:56 +08:00
Mashiro c458829763 Refine rfsearch and fix a typo 2023-03-20 18:26:52 +08:00
Hakjin Lee 1f9e5b57c2 [Fix] Fix a typo in the docstring of MSDeformAttn (#2599) 2023-03-20 18:26:52 +08:00
BainOuO b248efa5c3 Fix the paper link error in the docstring (#2618) 2023-03-20 18:26:52 +08:00
Hakjin Lee 4ad79591f9 [Enhancement] Support MultiScaleDeformableAttention with AMP (#2541)
* [Enhance] Support FP16 for MSDeformAttn

* [Fix] Data type mismatch

* Update mmcv/ops/multi_scale_deform_attn.py

* Add UT

Author:    nijkah <nijkah@gmail.com>

* Add cuda available condition

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-20 18:26:52 +08:00
enemy1205 0e14ce2c1d [Feature] Add C++ implementation for bbox_overlaps (#2477)
* add ops bbox_overlaps

* format code

* Return the pytorch version

* Intermediate modification

* Solve problems in parameter passing

* revise bug

* "add test case"
2023-03-20 18:26:52 +08:00
Shanghua Gao bc10b5f0c0 [Feature] Support receptive field search of CNN models (#2056)
* support rfsearch

* add labs for rfsearch

* format

* format

* add docstring and type hints

* clean code

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

* rm unused func

* update code

* update code

* update code

* update  details

* fix details

* support asymmetric kernel

* support asymmetric kernel

* Apply suggestions from code review

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

* Apply suggestions from code review

* Apply suggestions from code review

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

* Apply suggestions from code review

* Apply suggestions from code review

* Apply suggestions from code review

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

* Apply suggestions from code review

* add unit tests for rfsearch

* set device for Conv2dRFSearchOp

* Apply suggestions from code review

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

* remove unused function search_estimate_only

* move unit tests

* Update tests/test_cnn/test_rfsearch/test_operator.py

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

* Apply suggestions from code review

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

* Update mmcv/cnn/rfsearch/operator.py

Co-authored-by: Yue Zhou <592267829@qq.com>

* change logger

* Update mmcv/cnn/rfsearch/operator.py

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

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
Co-authored-by: lzyhha <819814373@qq.com>
Co-authored-by: Zhongyu Li <44114862+lzyhha@users.noreply.github.com>
Co-authored-by: Yue Zhou <592267829@qq.com>

[Fix] Fix skip_layer for RF-Next (#2489)

* judge skip_layer by fullname

* lint

* skip_layer first

* update unit test
2023-03-20 18:26:52 +08:00
Zaida Zhou e7adffb9e0
[Fix] Skip filtered_lrelu ut when cuda is less than 10.2 (#2677) 2023-03-19 16:17:45 +08:00
NoCakeAnyMore d31b22125f
[Fix] nms_rotated_npu accepts angle parameter (#2674) 2023-03-17 17:30:50 +08:00
q.yao 03ea1c9d99
[Fix] Skip fused_lrelu op when gcc is less than 6.0 or cuda is less than 10.2 (#2671)
* disable filtered_lrelu_op

* fix lint

* add cuda version check

* warning if disable
2023-03-17 17:29:35 +08:00
Zaida Zhou 91ed30dd53
Build macOS pre-built packages for PyTorch2.0 in CI (#2668)
* Add support for pytorch2.0

* minor fix

* update
2023-03-17 15:40:35 +08:00
liuhw 3ad02efde2
[Fix] Prevent divide-by-zero error on Ascend device for bbox_overlaps (#2646) 2023-03-15 23:08:15 +08:00
Yifei Yang 869dbf1bf2
[Feature] Add Ops of StyleGAN3 (#2290)
* add bias_act

* support bias_act

* support filtered_lrelu

* support filtered_lrelu and upfirdn2d

* support conv2d_gradfix and fix filtered_lrelu

* fix lint

* fix lint

* fix c++ lint

* fix part comments

* fix lint

* rm redundant header

* fix upgrade pip

* fix as comment

* fix c++ lint

* fix ci

* fix-ut

* fix as comments

* add grad check

* remove redundant template

* Update mmcv/ops/bias_act.py

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

* add typehint

* fix as comment:

* complete type hints

* fix lint

* add test for conv_gradfix

* add test for conv_gradfix

* fix lint

* modify licenses and ops.md

* add zh op md

* add torch version policy for conv2d_gradfix

* fix lint

* fix as comments

* rename impl

* rm redudant function and add ut

* fix as comment

* fix lint

* fix lint

* fix as comments

* fix lint

* fix ut

* fix as comment

* fix as comment

* fix as comment

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-03-13 16:05:11 +08:00
liuhw 0fb07d0e69
[Feature] Add the support of voxelization op for ascend device (#2614)
* Add voxelization op npu adaptor

* update

* update
2023-03-03 20:07:11 +08:00
q.yao 998e459769
[Fix] Call operator __half explicit (#2608) 2023-02-18 00:16:34 +08:00
Mashiro a019d2a3cc
Register Compose to TRANSFORMS (#2600) 2023-02-17 15:07:35 +08:00
ckirchhoff 834f94db23
[Feature] Add support of some ops for Ascend device (#2594)
* [Feature]: add supports of gather_points, nms_rotated, bbox_overlaps for Ascend device

* Apply suggestions from code review

---------

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-02-13 16:43:42 +08:00
Qian Zhao 8e2b2bf362
[Docs] Add api references in docs (#2587) 2023-02-08 20:37:41 +08:00
tripleMu c84d52cda4
Use official isort (#2575) 2023-01-31 12:59:09 +08:00
tripleMu 6cae40a45b
[Fix] Upgrade the version of isort to fix lint error (#2567) 2023-01-29 21:06:25 +08:00
Zaida Zhou eb57b3bbbe
Bump version to v2.0.0rc4 (#2559) 2023-01-18 10:25:36 +08:00
Zaida Zhou 009cfe602d
fileio related interfaces should keep bc (#2539)
* fileio related interfaces should keep bc

* fix format

* update unit tests of transforms

* update format

* lazily infer client until calling get function of fileio
2023-01-17 13:10:12 +08:00
ckirchhoff 34bdf448ac
[Feature] Pick npu ops from master to 2.x (#2501)
* merge npu ops from master to 2.x

* BugFix: fix merge bugs

* {[Feature]: add psamask, roipool to 2.x, and fix the SigmoidFocalLoss assert condition

* merge conflicts in ops.md

* [fix]: fix merge bug
2023-01-13 13:39:35 +08:00
liukuikun 7156604eba
[Feature] Add BezierAlign CUDA op (#2393)
* bezier align

* add ut

* fix comment

* updata ut

* fix link and comment

* fix comment
2023-01-10 17:05:39 +08:00
CokeDong 78fdf17817
Update parrots image (#2538)
* Update image for 2.x

* Update install cmd for 2.x
2023-01-10 11:02:59 +08:00
Mashiro 16155078f4
[Fix] Fix CI failed due to the version of protobuf (#2529)
* [Fix] Fix CI failed caused by protobuf

* minor refine

* test minimum_version_gpu

* test minimum_version_gpu

* test minimum_version_gpu

* fix ci

* remove protobuf from requirements

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2023-01-08 14:29:38 +08:00
Zaida Zhou fadd915cb8
[Fix] _flip_keypoint of RandomFlip returns a wrong result (#2527) 2023-01-06 20:21:58 +08:00
Mashiro 4ae327f43c
Deprecate the support for python3.6 (#2525)
* [Fix] Adjust the minimum supported version of Python to 3.6

* refine the docs
2023-01-06 15:39:44 +08:00
Zaida Zhou a3cb068590
[Refactor] Use the new interface of fileio from mmengine (#2468)
* [Refactor] Use new interface of fileio

* update ut

* remove file_client_args

* add * before backend_args

* refine backend_args description
2022-12-28 00:09:05 +08:00
q.yao 32cc2f7c72
[Fix] Fix the version of Windows OS to 2019 (#2503) 2022-12-20 22:50:57 +08:00
takuoko d510b8b174
[Feature] Support LayerScale in FFN (#2451)
* add layer scale

* add layer scale

* add layer scale

* Update mmcv/cnn/bricks/transformer.py

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

* Update mmcv/cnn/bricks/transformer.py

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

* add layer scale

* move LayerScale

* add layer_scale_init_value

* add typehint

* fix for tensor with any dim

* fix layer scale rule

* fix layer scale rule

* fix test

* add docs

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-12-11 17:58:19 +08:00
takuoko 433607030a
[Enhancement] Support value_proj_ratio in MultiScaleDeformableAttention (#2452)
* add ratio in ms_deform_attn_

* add ratio in ms_deform_attn

* Update mmcv/ops/multi_scale_deform_attn.py

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

* Update tests/test_ops/test_ms_deformable_attn.py

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

* add ratio in ms_deform_attn

* add ratio in ms_deform_attn

* add ratio in ms_deform_attn

* add ratio in ms_deform_attn

* add ratio in ms_deform_attn

* add ratio in ms_deform_attn

Co-authored-by: Zaida Zhou <58739961+zhouzaida@users.noreply.github.com>
2022-12-11 17:48:21 +08:00
vansin fb39e1e568
[Docs] Update qq group link (#2454)
* [Doc] update qq group link

* [Doc] update qq group link

* [Doc] update zhihu link

* [Doc] update the wechat qrcode link
2022-12-06 17:33:34 +08:00
liukuikun fd11b874cf
[Docs] Add convention for the order of data transforms (#2442) 2022-12-06 17:06:20 +08:00
Kei-Chi Tse 6e8183092a
[Fix] Fix optimizer 'ValueError' when using PReLU activation (#2444)
* fix optimizer ValueError when using PReLU activation.

* fix lint error
2022-12-06 16:57:58 +08:00
Zaida Zhou 2478cb6f84
[Docs] Update supported pytorch and cuda versions (#2456)
* Update supported pytorch versions

* update

* update
2022-11-28 10:59:07 +08:00
Zaida Zhou 104aaa8b78
Bump version to v2.0.0rc3 (#2447) 2022-11-23 18:55:21 +08:00
谢昕辰 b852c1c433
[Fix] Fix a not proper assertion in RandomChoiceResize (#2450)
* fix assertion

* remove assert

* use is_seq_of
2022-11-23 18:54:11 +08:00
ZShaopeng a0939977a3 [Feature] Support MultiScaleDeformableAttn with cambricon MLU backend 2022-11-23 16:57:35 +08:00
cathyzhang222 193de43bc8 [Feature] Support NMS with cambricon MLU590 backend (#2401)
* [Feature] Support Nms with cambricon MLU590 backend

support 590 for nms

* add blank
2022-11-23 16:57:35 +08:00
bdf e847cf8ad4 [Refactor] Adapt mlu code to cntoolkit3.0.1 2022-11-23 16:57:35 +08:00
Zaida Zhou 4c6e99c8e1 [Docs] Add code style guide (#2406)
* [Docs] Add code style guide

* add typing and mypy links

* fix format
2022-11-23 16:57:35 +08:00
Mashiro 7e7425f941 [Docs] Refine contributing.md (#1941)
* refaictor CONTRIBUDING.MD

* refaictor CONTRIBUDING.MD

* fix source rvm shell command

* fix lint

* add chinese contributing.md

* fix error render

* fix error render

* refine description

* refine description

* minor refine

* install ffmpeg before ut

* Minor retfine

* Minor refine

* refine chinese contributing.md

* Refine

* Fix as comment

* Fix as comment

* restore pr.md

* minor refine

* minor refine

* update chinese CONTRIBUTING.md in root dir

* adjust imgae size

* update en CONTRIBUTING.md

* update en CONTRIBUTING.md

* update contributing.md in docs

* restore change of index.rst

* Fix as comment

* minor refine

* remove content in PR.md

* minor refine

* minor refine

* minor refine

* Fix as comment

* rename

* minor refine
2022-11-23 16:57:35 +08:00
q.yao b0646e0ec9 [Fix] Fix Correlation op (#2274)
* fix correlation

* fix lint
2022-11-23 16:57:35 +08:00
cathyzhang222 6c89b71756 [Feature] Support Iou3d with cambricon MLU backend (#2339)
* [Feature] Support Iou3d with cambricon MLU backend

* [Feature] add double line

* [Feature] add line
2022-11-23 16:57:35 +08:00