Commit Graph

420 Commits (9de04d228455e490d9bcbb4fd0186a2b058d938a)
 

Author SHA1 Message Date
Jerry Jiarui XU 45a39dc6a8
make config merge a into b non-inplace (#254) 2020-04-29 15:09:27 +08:00
Kai Chen f7de63fdd0
Support specifying names for registry.register_module() (#251)
* support specifying names for registry.register_module()

* minor fix

* add more unittests
2020-04-27 22:47:49 +08:00
Harry c203419f57
fix: deadlock if processes have different log_buffer (#252) 2020-04-27 22:44:26 +08:00
Rui Xu 0c34ab90d0
fix previous bug in save-model (#250)
* flush buffer when saving model

* fix bugs

Co-authored-by: xurui <xurui4@sensetime.com>
2020-04-27 17:27:30 +08:00
Wenwei Zhang d6210e6d51
Fix cyclic registry (#249)
* track progress of iter&enum

* restore

* Add CyclicLrUpdater in Hooks
2020-04-27 13:35:11 +08:00
Kai Chen a338d43d78
Refactor unittests (#241)
* refactor unittests

* split test_video.py to two files
2020-04-26 22:54:27 +08:00
Cao Yuhang b7e8d7d7fb
Use f-string (#245)
* use f-string

* delete python3.5 sup

* minor fix

* fix supported python version

* fix format

* fix yapf

* remove redundant space

* fix typo
2020-04-26 00:21:54 +08:00
Jerry Jiarui XU 1e8a212187
Add DictAction and docs for config (#243)
* fixed merge_from_dict, add DictAction

* add config docs

* fixed format type

* change to easy example

* update docs

* update docs
2020-04-25 20:56:08 +08:00
Maxim Bonnaerens 76a064ee05
Support optimizers with no momentum (#240)
* Support optimizers with no momentum

* put else-if outside return
2020-04-25 18:17:58 +08:00
Kai Chen e333d8222b
Add FileClient to access file from different backends (#237)
* add file client

* install missing requirements

* use .format() instead of f-string
2020-04-23 19:54:29 +08:00
Kai Chen a0618d1051
Refactoring mmcv.images (#239)
* refactoring mmcv.images

* update docstring and minor fix

* some renames
2020-04-23 00:34:51 +08:00
Kai Chen 010b1a0ffc
remove parallel_test (#238) 2020-04-22 23:47:44 +08:00
Kai Chen af02ac9f01
Use tmp dir as work_dir of runner (#236)
* use tmp dir as work_dir of runner

* only run codecov for python 3.7

* remove useless comments
2020-04-22 23:33:54 +08:00
Kai Chen ad98674fa3
Update to v0.4.4 (#235) 2020-04-22 19:42:28 +08:00
Cao Yuhang 887ad81e70
add resnet msra caffe (#231) 2020-04-21 17:03:47 +08:00
Wenwei Zhang ba059611d1
Momentum scheduler (#167)
* track progress of iter&enum

* restore

* add momentum scheduler

* fix small bug

* cyclic scheduler"

* fix bug

* fix second phase's bug

* reformat

* feature (cosine lr): use relative ratio for more flexible scheduler

* Fix (runner): fix bugs in runner

* Refactor (hook): refactor cosing/cyclic LR/momentum hook with unittest

* Clean unnecessary files and reformat

* Fix memory key error when GPU is not avaliable

* Resolve comments

* Do not print momentum in text log

* Change hook register order

* Refactor max_iter

* Fix max_iter bugs in runner

* Enforce target_ratio to be either tuple or float
2020-04-20 01:23:53 +08:00
Bin Zhang 8ac858b138
add max_keep_ckpts parameter (#227)
* add max_keep_ckpts parameters to save memory when saving models

* format

* format

* format

* fixed linting error

Co-authored-by: z-bingo <z-bingo@outlook.com>
2020-04-17 16:01:30 +08:00
Rui Xu 732c379761
flush buffer when saving model (#232)
Co-authored-by: xurui <xurui4@sensetime.com>
2020-04-17 13:27:29 +08:00
David de la Iglesia Castro d5f190d12d
Add MlflowLoggerHook (#221)
* Add MLflowLoggerHook

* Add MLflowLoggerHook to __all__

* Update name

* Fix tracking.MlflowClient setup

* Fix log_metric

* Fix mlflow_pytorch import

* Handle active_run

* Fix self.mlflow reference

* Simplify using high level API

* Fix set_experiment

* Add only_if_torch_available decorator and test_mlflow_hook

* Add missing import in hooks

* Fix torch available check

* Patch mlflow.pytorch in test

* Parametrize log_model

* Fix log_model parametrize

* Add docstring

* Move wand patch

* Fix flake8

* Add regression test for non numeric metric

* Only log numbers

* Rename experiment_name-> exp_name

* Remove pytest skip
2020-04-14 23:54:55 +08:00
Kai Chen 728b88dfdb
update CI config and pre-commit hook config (#228) 2020-04-08 00:04:50 +08:00
Kai Chen 25a21d7b2b
bump version to v0.4.3 (#226) 2020-04-07 19:03:07 +08:00
Joanna 2ea0d51816
add contains to registry (#223)
* add contains to registry

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

* add contains to registry

* 'fix'

* 'minor'

* move

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

Co-authored-by: lixuanyi <lixuanyi@sensetime.com>
2020-04-06 12:49:13 +08:00
Jerry Jiarui XU e87a04e3c7
add pretty_text (#220)
* add pretty_text

* add test cases

* add argument indent

* change pretty_text to property

* fixed format end
2020-04-03 21:18:34 +08:00
lizz 9895a1af80
Check empty before clip grads (#219)
Signed-off-by: lizz <lizz@sensetime.com>
2020-04-02 22:49:54 +08:00
Kai Chen cd873f9c2f
Fix a potential bug of removing temp dir (#218)
* fix a potential bug of removing temp dir

* bump version to v0.4.2
2020-03-28 22:00:19 +08:00
Xintao 1bc80610ee
support recursive scanndir (#215)
* recursive scanndir

* support python3.5

* update and add tests

* resolve comments

* remove is_str
2020-03-19 22:50:37 +08:00
Jerry Jiarui XU fa84b26422
add bias_init_with_prob (#213)
* add bia_init_with_prob

* add test_weight_init
2020-03-17 20:48:46 +08:00
Kai Chen 12e5913bb9
Remove supports for python 2.7 and bump version to 0.4.0 (#211)
* remove supports for python 2.7

* fix the unit test for python 3.5

* add python 3.8 in CI

* try ubuntu 18.04 as the environment
2020-03-16 15:15:35 +08:00
Haodong Duan 2624090ff4
Warmup byepoch (#209)
* add iminvert

* DC now supports dimension other than 3

* add lmdb

* revise for lmdb

* add warmup_byepoch option

* remove not needed files

* add attribute active_loader for runner, revise code

* remove not used attrs

* remove not used attrs

* add docstring, else

* to pass CI

* to pass CI

Co-authored-by: zhaoyue-zephyrus <thuzhaoyue@gmail.com>
2020-03-15 22:14:39 +08:00
Jerry Jiarui XU 6738cd3085
add base for config (#194)
* add base for config

* fixed format

* rm terminal width

* support multiple & recursive base

* add test case

* fix format

* add test construct

* minor fix

* add more test, rewrite merge from opt

* avoid depulicate keys

* delete imported config as module

* rename merge_from_dict
2020-03-15 20:29:17 +08:00
Rui Xu 7dac84d08c
support pathlib in imageio (#208)
* support pathlib in imageio

* support pathlib in imageio

* add Path type for the annotation.

Co-authored-by: xurui4 <xurui4@sensetime.com>
2020-03-14 23:16:44 +08:00
lizz c294d2713b
Support dot in config.py file (#207)
Signed-off-by: lizz <lizz@sensetime.com>
2020-03-12 23:20:50 +08:00
NightBaronStar 24a66f1d47
Update tensorboard.py (#205)
* Update tensorboard.py

* Update tensorboard.py
2020-03-11 17:01:52 +08:00
Jiangmiao Pang fd066152d0
Update PaviLoggerHook (#202)
* add is_scalar in PaviLoggerHook and add unittest

* add docstrings
2020-03-02 18:48:51 +08:00
Kai Chen 120c6a64da
remove support for opencv 2 (#201) 2020-03-01 15:53:10 +08:00
Kai Chen 3fa5851891
Update pre-commit hook config (#200)
* update pre-commit hook config

* update the contributing guide
2020-03-01 15:51:40 +08:00
Jiangmiao Pang ac78bdc2cf
Add PaviLoggerHook (#198)
* fix wandb hook

* add pavi logger hook

* modify default value of init_kwargs to None

* add more features to PaviLoggerHook

* rm taskid (will cause bug)

* directly upload saved pth

* fix CI

* register PAVI Hooks
2020-03-01 00:39:16 +08:00
Kai Chen 0d5332a4c9
use registry to manage hooks (#199) 2020-02-29 20:46:40 +08:00
Kai Chen c2c9fced2b
Add logging utils (#196)
* add logging utils

* install torch (cpu only) for CI

* fix unittests for runner

* remove python 2.7 from testing list

* add a corner case
2020-02-24 22:31:36 +08:00
Kai Chen 34197c5c89
Add registry and build_from_cfg (#195)
* add registry and build_from_cfg

* add some corner cases

* add some corner cases

* fix the unittest for python 3.5

* minor fix
2020-02-24 22:21:01 +08:00
Kai Chen 0863073cb7
flatten the mmcv/image dir and update to v0.3.2 (#197) 2020-02-24 22:13:50 +08:00
Joanna 0bf1e80c6e
add fast imresize (#193)
* size hint

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

* add fast resize

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

* fix

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

* add tests

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

* fix

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

* Update tests/test_image.py

* change format

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

* add tests

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

* add docstring

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

Co-authored-by: lizz <innerlee@users.noreply.github.com>
2020-02-24 21:50:33 +08:00
Joanna 9332a1dd29
add inplace normalization (#192)
* add inplace normalization

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

* fix

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

* add dtype check

* modify inplace norm

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

* fix

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

* fix

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

* fix line
2020-02-19 22:49:15 +08:00
Joanna 5ee538d4ac
add inplace imflip (#191)
* add implace flip

* add tests

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

* modify tests

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

* add inplace tests

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

* add comments

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

* fix

Signed-off-by: lixuanyi <lixuanyi@sensetime.com>
2020-02-19 16:02:48 +08:00
Joanna 4eca2c5964
Image decoding optimization (#187)
* change image decoding method from cv2 to TurboJPEG

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

* add tests

* travis

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

* fix

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

* more doc

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

* change per request

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

* more doc

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

* baby-sitting

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

Co-authored-by: lizz <innerlee@users.noreply.github.com>
2020-02-18 20:41:03 +08:00
lizz a0506ec560
Use cv2 to normalize img (#176)
* Use cv2 to normalize img

* multipy faster than divide

* also do imdenormalize

* support more channels

* even faster

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

* add test

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

* even even faster

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

* fffaster

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

* okay

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

* guard against implace modification

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

* fix imdenormalize

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

* tidy

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

* clean

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

* Update test_image.py

* Update test_image.py

* Update test_image.py

* Update test_image.py

* Update test_image.py

* Update test_image.py
2020-02-15 23:59:54 +08:00
Kai Chen 9470ff7035
remove support for python 3.4 (#189) 2020-02-15 15:09:36 +08:00
Kai Chen 6fe192fcaa
update version to 0.3.0 (#188) 2020-02-15 14:04:27 +08:00
Kai Chen a47ca7a9a2
Fix an import error caused by legacy codes (#186)
* fix an import error

* fix an import error
2020-02-14 00:57:37 +08:00
Kai Chen f4272a8875
remove the outdated PaviLoggerHook and add requirements.txt (#185) 2020-02-12 22:46:12 +08:00