mirrors
/
mmcv
mirror of https://github.com/open-mmlab/mmcv.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
377 Commits
13 Branches
108 Tags
42 MiB
Commit Graph

13 Commits (039e06e43a69cb5959585d847087848e41b8eb09)

Author SHA1 Message Date
Jerry Jiarui XU 42f03d84de
Refactor plugins: move from ops to cnn (#380) 
* Refator plugins: move from ops to cnn

* minior update

* minior update
 2020-07-03 01:37:58 +08:00
louzana 2c6fc5fd9b
fix bug of building ConvModule with HSigmoid using inplace=True and a… (#369) 
* fix bug of building ConvModule with HSigmoid using inplace=True and add corresponding unittest

* fix linting
 2020-06-28 23:27:14 +08:00
zhuyuanhao c0f5492ee9
add ext ops, support parrots (#310) 
* add ext ops, support parrots

* fix lint

* fix lint

* update op from mmdetection

* support non-pytorch env

* fix import bug

* test not import mmcv.op

* rename mmcv.op to mmcv.ops

* fix compile warning

* 1. fix syncbn warning in pytorch 1.5
2. support only cpu compile
3. add point_sample from mmdet

* fix text bug

* update docstrings

* fix line endings

* minor updates

* remove non_local from ops

* bug fix for nonlocal2d

* rename ops_ext to _ext and _ext to _flow_warp_ext

* update the doc

* try clang-format github action

* fix github action

* add ops to api.rst

* fix cpp format

* fix clang format issues

* remove .clang-format

Co-authored-by: Kai Chen <chenkaidev@gmail.com>
 2020-06-28 23:15:47 +08:00
Jintao Lin 9bc0b96d02
Add utils to calculate model complexity and adjust folder structure for mmcv/cnn/utils . (#366) 
* Add utils to calculate model complexity info

* remove _InstanceNorm in unittest

* add docstring and increase unittest coverage

* fix deconv_flops_counter_hook to accept different data shape

* test when model is not a common instance

* put flops_counter.py and weight_init.py into mmcv/cnn/utils folder

* fix import name

* reformat some docstrings

* update the scripts with latest one and remove redundant codes

* directly represent a model without string and eval()

* reformat code
 2020-06-26 21:38:29 +08:00
louzana fef7b2fea4
add hswish and hsigmoid (#358) 
* add hswish and hsigmoid

* fix linting

* fix linting

* delete useless staticmethod in cnn/bricks/hsigmoid.py

* forward(x) ==> forward(seld, x)

* add unittests for hsigmoid and hswish

* use torch.equal instead of numpy.equal
 2020-06-19 21:35:17 +08:00
Jintao Lin dcc20f3a4b
Add `_NonLocalNd` module (#331) 
* add non_local module

* rewrite non local module comments

* perfect docstring and adjust init function

* not to init norm layer

* Correct initialize when there is a norm

* set normal method for both embedded_gaussian and dot_product
 2020-06-18 20:56:48 +08:00
Jerry Jiarui XU 2aa7712cb7
add more act w.o. inplace opt (#329) 2020-06-10 15:42:09 +08:00
Harry efecf7d1fe
fix building ConvModule with Tanh activation (#309) 
* fix: tanh has not inplace

* feat: add ConvModule with Tanh
 2020-06-01 21:26:51 +08:00
Jintao Lin 25d07e3d54
Support conv layers' own `init_weights` method (#278) 
* Support conv layers' own `init_weights` method

* Add related unittest about ConvModule init_weight

* make the comments more specific
 2020-05-13 22:58:16 +08:00
Wenwei Zhang c35c228cbb
Support to pass args to cnn bricks (#277) 2020-05-12 10:41:11 +08:00
Kai Chen d4fac3a6dd
Use parrots wrapper (#272) 
* use parrots wrapper for norms

* add unittests for is_norm()

* add a test case
 2020-05-08 11:24:35 +08:00
Kai Chen 6aa5131559
move test_weight_init to tests/test_cnn (#270) 2020-05-05 21:03:12 +08:00
Kai Chen 45111e193d
Add building bricks of cnn (#247) 
* add building bricks of cnn

* add unit tests

* use registry for building bricks

* minor updates

* add scale layer

* add test for scale

* add doc string

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
 2020-05-01 00:32:25 +08:00