Commit Graph

413 Commits (b20d6ab18b03dfe31515e8c54cb91efa9d82e451)
 

Author SHA1 Message Date
Jiamin b20d6ab18b
Update jenkinsfile (#444)
* chore: use https and rm work_dir before cp

* chore: explictly install Cython

* chore: rm dir:tmp before cp
2020-07-24 19:01:35 +08:00
Wang Xinjiang 66604e83de
Add syncbuffer hook (#443)
* reformat

* reformat

* Add register hook from cfg

* docstring

* change according to comments
2020-07-24 14:15:44 +08:00
robin Han 903091effc
add __init__.py so that we can find onnx_util while packaging (#442) 2020-07-24 12:00:05 +08:00
Jiamin fad12cd366
Add jenkinsfile (#415)
* chore: add jenkins file

* chore: remove cuda_arch

* chore: replace Float.parsefloat with string check

* chore: add pip mirror

* chore: fix add MMCV_WITH_OPS env var

* chore: fix undefined symbol _Z16THPVariable_WrapN2at6TensorE

* chore: add comment

* chore: fix multi docker running under same dir error

* chore: rm dir after build

* chore: fix rm

* chore: also rm tmp dir

* chore: use sensetime docker

* feat: use torch-pre-installed docker

* fix: fix index error
2020-07-23 17:06:34 +08:00
Kai Chen 487f381f25
fix docs (#440) 2020-07-21 14:48:16 +08:00
Jiamin 55fadb4c4e
Add runner.meta to checkpoint in save_checkpoint() (#438)
* fix: error when runner.meta is None

* tests: add unittest for epoch-based save_checkpoint
2020-07-20 11:40:04 +08:00
Yang Li 215a324414
Support update specific steps (#437) 2020-07-20 11:36:00 +08:00
sandyhsia 82e5c8e552
fix: close temp file after creating it to avoid permssion problem on windows (#436) 2020-07-20 00:16:39 +08:00
Kai Chen 3f936167de
bump version to 1.0.3 (#434) 2020-07-17 23:48:59 +08:00
Yawei Li 7730a79fcd
fix typo of annealing (#433) 2020-07-17 23:48:22 +08:00
Jintao Lin 6ece0e5d19
Add pairwise function for 'gaussian' and 'concatenation' mode in NonLocal. (#424)
* add pairwise function for 'gaussian' and 'concatenation' mode

* rename test function

* decrease the complexity of nonlocal unittest

* fix typo and make unittest more complete

* add unittest when zero_init is False

* minor fix

* pack theta and phi

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-07-17 23:44:50 +08:00
Jerry Jiarui XU 49fdf3cfa0
add FP16 optimizer (#432) 2020-07-17 21:25:47 +08:00
Jerry Jiarui XU 9e3ff5f94c
Support pillow resize (#426)
* Support pillow resize

* add more types
2020-07-17 19:25:47 +08:00
robin Han 34ee69e0f6
support converting pytorch to onnx (#429)
* support convert nms to onnx op

* add comment for NMSop

* add corresponding onnx support for mmdet/mmseg/mmediting

* remove onnx part from mmcv/__init__
2020-07-17 19:24:46 +08:00
Jerry Jiarui XU 07e4215286
Fixed group of syncbn (#427) 2020-07-17 11:16:47 +08:00
Jerry Jiarui XU 4a044c6466
Support FP16 (#428)
* Support FP16

* update ci
2020-07-17 11:14:07 +08:00
Jerry Jiarui XU 7cfc839ea5
Add MMSyncBN in registry (#420)
* Add MMSyncBN in registery

* skip mmsyncbn test
2020-07-15 19:23:09 +08:00
Jiamin 50f69e7057
fix: _get_max_memory when model has no attr output_device (#416) 2020-07-15 17:37:14 +08:00
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