Commit Graph

495 Commits (ed2887bb72b7c8cee11aba94663831b8f6bd8827)
 

Author SHA1 Message Date
Rui Xu 71e327494e
[Bug]: return directly from running function (#413)
* fix bug of return directly from run function

* fix epoch based runner

* remove debug print
2020-07-14 15:10:40 +08:00
su 123bc9b03c
[feat]add error log in config validate check (#414) 2020-07-14 14:45:33 +08:00
Wenwei Zhang 6cf337374c
Add more projects in readme (#410)
* add more projects in readme

* fix typo

* Add mmaction2

* reformat
2020-07-13 00:53:30 +08:00
Wenwei Zhang 8b62aa60ed
use latest rather than 1.0.0 for mmcv installation (#408)
* use latest rather than 1.0.0 for mmcv installation

* Fix install bug
2020-07-11 22:28:07 +08:00
Cao Yuhang 82a15dfcef
compitable dcn old api (#405) 2020-07-10 00:14:28 +08:00
shilong 481fdbf30b
fix a bug in some opencv version (#407) 2020-07-10 00:14:14 +08:00
Cao Yuhang 2b876bc7a7
Compitable old interface (#404)
* compitable-old-interface

* compitable softnms

* rename

* rename

* use warnings

* upgrade version to 1.0.1
2020-07-09 21:31:08 +08:00
Kai Chen cc70cf8ec3
bump version to 1.0.0 (#402) 2020-07-09 02:57:38 +08:00
Jerry Jiarui XU da7a42791b
Bump to rc2 (#401)
* Bump to rc2

* change rc2 -> 1.0.0
2020-07-09 02:39:48 +08:00
Jerry Jiarui XU 192b3b3ceb
use env to control compile op or not (#400)
* use env to control compile op or not

* use str as env var

* update readme

* update rc01

* update readme

Co-authored-by: Kai Chen <chenkaidev@gmail.com>
2020-07-09 02:19:02 +08:00
Harry 5704613e28
Remove all module wrapper's module when saving checkpoint (#399)
* fix: remove all module wrapper when saving checkpoint

* refactor: move position of if

* docs: add docstring

* refactor: add _save_to_state_dict from official torch

* docs: modify docstring of _save_to_state_dict

* docs: modify docstring

* feat: add unittest

* feat: add DataParallel to unittest

* fix: a bug when model has batchnorm

* docs: update docstring
2020-07-08 23:20:22 +08:00
Joanna 27cc439d01
Update config to support predefined variables (#348)
* update config with predefined variables

* rm redun

Signed-off-by: lixuanyi <lixuanyi@sensetime.com>

* add test for config

Signed-off-by: lixuanyi <lixuanyi@sensetime.com>

* support all types

Signed-off-by: lixuanyi <lixuanyi@sensetime.com>

* newline at the end

Signed-off-by: lixuanyi <lixuanyi@sensetime.com>

* update

Signed-off-by: lixuanyi <lixuanyi@sensetime.com>

* extract code into a function and add docs

Signed-off-by: lixuanyi <lixuanyi@sensetime.com>

* fix and add tests

Signed-off-by: lixuanyi <lixuanyi@sensetime.com>

* add unit tests and fix

* fix

* fix minor

* fix test
2020-07-08 20:53:54 +08:00
Cao Yuhang 926ac07bb8
Move fuse conv bn to mmcv (#382)
* move fuse conv bn to mmcv

* update doc

* update test conv bn

* rename

* fix doc and variable name

* change func name
2020-07-08 19:03:15 +08:00
Cao Yuhang d678986289
fix mem cal bug (#398) 2020-07-08 19:02:36 +08:00
Cao Yuhang d5cbf7eed1
Migrate op (#392)
* migrate op

* migrate unittest

* update build no torch

* add back use_torch_vision for roi align

* fix type and unit test

* ignore test logging when no torch

* fix no torch ci test

* skip test registry

* remove coverage report when no torch

* fix mac ci order

* install latest pillow when no torch

* mv convws to brisk
2020-07-08 17:29:15 +08:00
Harry 33ca908529
feat: add mmediting pretrain weight to open_mmlab model_zoo (#395) 2020-07-08 15:27:22 +08:00
Wenwei Zhang 01bce661d2
Fix the path issue of logo (#394) 2020-07-08 12:47:37 +08:00
Wenwei Zhang 18d6372027
Rewrite readme.rst by markdown (#391)
* Rewrite readme.rst by markdown

* Update setup.py

* Update mmcv links

* clean code

* Add logo

* Resolve comments

* clean unnecessary code change
2020-07-08 01:20:27 +08:00
tianyuandu 039e06e43a
use torch.cummax in corner_pool for torch 1.5+ (#390)
* synchronize from mmdetection

* fix parrots
2020-07-07 17:28:32 +08:00
Kai Chen b11c56603f
fix syncbn when dist is unavailabel (#388) 2020-07-07 15:04:50 +08:00
Kai Chen b2b42cbd95
Add macos building in github action (#387)
* add macos in github action

* fix a package name

* minor fix

* fix typo

* ignore progressbar testing

* ignore timer testing

* fix pytest configs

* try fix
2020-07-07 13:14:24 +08:00
Yuanhao Zhu b18a33833e
fix macOS compile (#386) 2020-07-07 01:00:37 +08:00
Kai Chen e2ee171a4a
Fix the version of isort (#385) 2020-07-05 11:12:02 +08:00
Kai Chen 4004a3d7b6
update pre-commit hook config (#384) 2020-07-04 02:53:51 +08:00
Kai Chen 63b7aa31b6
Fix docstring formats (#383)
* update doc formats

* update docstring
2020-07-04 00:55:25 +08:00
Jerry Jiarui XU a47451b47a
add more module in cnn __init__ (#381) 2020-07-03 11:49:47 +08:00
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
Xiaojie Li 82211b407e
Added new features to `impad` (#377)
* update impad

* fix docstring

* add shape for impad

* fix unit test

* remove old version & fix doc

* fix linting

* fix doc

Co-authored-by: lixiaojie <lixiaojie@sensetime.com>
2020-07-02 19:34:46 +08:00
Kai Chen cab4031454
include cuda and cpp files (#379) 2020-07-02 19:18:22 +08:00
Kai Chen 3777aa8c10
Fix the version of codecov-action to v1.0.10 (#378)
* Fix the version of codecov-action to v1.0.7

* Update build.yml
2020-07-02 14:41:58 +08:00
Jerry Jiarui XU 9c51928b7c
Add tensor2imgs in misc (#374)
* add tensor2imgs in misc

* rename

* fixed import

* raise error
2020-07-02 00:14:50 +08:00
Jintao Lin 1ebd7ea6fb
add unittest for set_random_seed (#376) 2020-07-02 00:13:04 +08:00
Jerry Jiarui XU 0970ae94c2
Add cpu ops (#370)
* update actiion

* remove 1.3 cuda

* reorder

* add no torch

* fixed version

* make py3.8 on pt1.5

* make py3.8 on pt1.5

* remove torch 1.3

* disable py3.8

* pip

* merge master with cuda compile fix

* add cpu roi align

* fixed test

* fixed no torch

* add CUDA_ARGS

* use one line

* gencode=61

* seperate jobs

* update lint

* use parametrize test

* formart and rename

* unit test for all
2020-06-30 12:09:48 +08:00
Rui Xu ab77af82d6
add set_random_seed and use rank shift feature (#373) 2020-06-29 22:26:58 +08:00
zhuyuanhao e43fe0e243
remove cuda args (#372) 2020-06-29 21:14:09 +08:00
zhuyuanhao d9549fba04
fix cpp header error (#371)
* 1. use macro USE_PARROTS control header include
2. add clang-format google style in pre-commit

* use MMCV_ macros
2020-06-29 18:48:50 +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
Kai Chen a7bf770168
bump version to v0.6.2 (#368) 2020-06-26 21:46:18 +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
Y. Xiong fcced0f0f8
Lite version: use mmcv without installing pytorch as a requirement (#367)
* use mmcv without installing pytorch

* add new line

* revise utils.init by try-except-else block

* add new line to utils.init

* del whitespace in new line

* add flake8 ignore to utils.__init__
2020-06-25 17:42:44 +08:00
Harry 69048ff056
Specifying distributed training port in os.environ when training with slurm (#362)
* feat: support for os.environ port for slurm training

* fix: port data type

* feat: add flawed unittest

* feat: add flawed unittest

* docs: add comments

* fix: unittest

* fix: unittest
2020-06-20 00:49:44 +08:00
Jerry Jiarui XU 8e80223e93
add more HRNet (#360) 2020-06-19 22:33:45 +08:00
Kai Chen 17d2aa5bc6
Bump version to v0.6.1 (#361) 2020-06-19 21:51:04 +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
Jerry Jiarui XU e3d9bf42c5
add resnetv1c (#359) 2020-06-19 18:07:44 +08:00
Harry 68708cfbe6
fix: update module wrapper register_module (#357) 2020-06-19 00:45:46 +08:00
Kai Chen 12aab3e19e
bump version to 0.6.0 (#356) 2020-06-18 22:15:22 +08:00
Harry 4808988117
Fix potential bug in DefaultOptimizerConstructor (#355)
* fix: fix custom_keys whenweight_decay is not specified

* docs: small fix

* feat: add type check

* refactor: move if outside for
2020-06-18 22:07:16 +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