mirrors/mmcv
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmcv.git synced 2025-06-03 21:54:52 +08:00
Code Issues Projects Releases Wiki Activity
1,348 Commits 13 Branches 108 Tags
Commit Graph

482 Commits

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
a6d7dde77f
[Fix] Fix torch2.0 dcn/mdcn symbolic (#2695) 
* fix

* fix lint
 2023-03-24 13:01:56 +08:00
Mashiro
c458829763 Refine rfsearch and fix a typo 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
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
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
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
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
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
ZShaopeng
a0939977a3 [Feature] Support MultiScaleDeformableAttn with cambricon MLU backend 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
Zhang
e9777adbac [Feature] Support RoIAwarePool3d with cambricon MLU backend (#2359) 
* [Feature] Support RoiawarePool3d with cambricon MLU backend.

* [Feature] Support RoiawarePool3d with cambricon MLU backend.
 2022-11-23 16:57:35 +08:00
q.yao
2d10616b8f [Fix] Fix three nn op can not accept half tensor (#2348) 
* Fix three nn half inpt

* update test
 2022-11-23 16:57:35 +08:00
q.yao
3bb0611aef [Fix] Create Tensor with new_* method to support amp (#2389) 2022-11-23 16:57:35 +08:00
VVsssssk
4e0341f26f
[Features] Add stack ball query and stack group points ops (#2292) 
* add stack sa model ops

* fix lint

* fix lint

* fix comments

* fix bug

* fix lint

* fix comments

* fix lint

* fix lint

* fix
 2022-10-28 00:42:10 +08:00
ZShaopeng
a8f7ae48e2 [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-10-22 13:39:50 +08:00
bdf
a364e6cad2 [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-10-22 13:39:50 +08:00
CokeDong
8fae3b4682 [Feature] Support PrRoI op for Parrots (#2280) 
* Support parrots extension for op PrRoI

* Fix lint

* Fix cpp lint

* Fix testcase failure by false requires_grad in self-defined autograd Funtion

* Fix issues

* Fix flake8

* Fix isort

* Adaption for typechecking for PrRoIPoolFunction

* Fix lint

* Support only float32

* bugfix
 2022-10-22 13:39:50 +08:00
takuoko
d8ac46df59 [Enhancement] Support SiLU with torch < 1.7.0 2022-10-22 13:39:50 +08:00
takuoko
dfef152989 [Feature] Register a new activatation layer SiLU to ACTIVATION_LAYERS 2022-10-22 13:39:50 +08:00
q.yao
e0b3223baf fix ci 2022-10-22 13:39:50 +08:00
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