807 Commits

Author SHA1 Message Date
Kai Chen
ad98674fa3
Update to v0.4.4 (#235) v0.4.4 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) v0.4.3 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
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
v0.4.1 v0.4.0
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
v0.3.2
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) v0.3.1 2020-02-15 15:09:36 +08:00
Kai Chen
6fe192fcaa
update version to 0.3.0 (#188) v0.3.0 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
Cao Yuhang
a500a64621
Use official ddp to implement MMDDP (#184) 2020-02-12 22:20:04 +08:00
Cao Yuhang
89606a14fa
log meta (#183)
* log meta

* add type check
2020-02-12 21:43:38 +08:00
Kai Chen
e127ef28db
Fix the missing numpy module for readthedocs (#182)
* Fix the missing numpy module for readthedocs

* fix the linting
2020-02-12 17:12:38 +08:00
lizz
0874642fb0
Speed up bbox_clip (#177)
* Speed up bbox_clip

* Update geometry.py
2020-02-05 15:17:42 +08:00
Kai Chen
d6ce3a2e1b
add _load_checkpoint() to handle the loading only (#178) 2020-02-03 23:44:39 +08:00
lizz
712651aab6
Adaptive progress bar length (#174)
* Adaptive progress bar length

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

* it works

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

* format

* pass test

* :lipstick

* test

* Update test_progressbar.py

* 2.7

* sort import

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

* try this

Signed-off-by: lizz <lizz@sensetime.com>
2020-02-02 18:18:17 +08:00
lizz
90a0f353ee
Format to conform pre-commit (#173)
Signed-off-by: lizz <lizz@sensetime.com>
2020-02-01 10:14:55 +08:00
Kai Chen
43ca0e57a9
fix the shape assertion in impad (#172) v0.2.16 2020-01-11 17:53:35 +08:00
Kai Chen
10fa1eeaf4
Add copyright header (#171)
* add copyright header

* change # to // for cpp files
2020-01-10 13:34:42 +08:00
Wenwei Zhang
da7bb063d3 Load state dict (#164)
* use load_from_state_dict to load ckpt

* reformat

* reformat

* reformat import

* pass flake8

* isort skip
2020-01-04 14:55:01 +08:00
Kevin
b34eef6e0e fix ZeroDivisionError in progressbar (#162) (#163) 2020-01-02 15:23:42 +08:00
Ye Liu
0d00f1a61c Fix int support in mmcv.slice_list (#161) 2019-12-21 19:34:08 +08:00
Kai Chen
ead4bc39bb
Add dist utils (#157)
* add dist utils

* rename due to a typo

* bump version to 0.2.15

* fix imports
v0.2.15
2019-12-12 22:18:40 +08:00