Commit Graph

420 Commits (9de04d228455e490d9bcbb4fd0186a2b058d938a)
 

Author SHA1 Message Date
Jintao Lin 9de04d2284
Replace `re` with `regex` (#457)
* replace `re` with `regex`

* add regex in requirements

* use regex on windows and re on linux

* update requirements.txt
2020-08-03 11:45:50 +08:00
Jerry Jiarui XU 66864d4abc
Update pretrain link (#455)
* update pretrain link

* remove some
2020-08-02 22:00:05 +08:00
shilong 1830347f8b
Ema (#421)
* add ema hook

* add ema hook resume

* add ema hook test

* fix typo

* fix according to comment

* delete logger

* fix according to comment

* fix unitest

* fix typo

* fix according to comment

* change to resume_from

* typo

* fix isort
2020-07-30 22:06:19 +08:00
Cao Yuhang 4676031c8b
Move onnx op import inside symbolic (#453)
* add pat

* move onnx op import to symbolic
2020-07-28 23:14:36 +08:00
Jintao Lin 829d1e381d
add function and class related with fp16 to runner/__init__.py (#448) 2020-07-27 10:43:20 +08:00
Wang Xinjiang d4da3daa7e
Syncbuf (#447)
* More robust sync buffer hook

* More robust sync buffer hook

* Reformat
2020-07-25 12:51:46 +08:00
Kai Chen cec5aacefb
bump version to 1.0.4 (#446) 2020-07-24 22:03:15 +08:00
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