Commit Graph

99 Commits (799421003d3f88e9b10a13c9ead0a3f7f84688d0)

Author SHA1 Message Date
lizz c294d2713b
Support dot in config.py file (#207)
Signed-off-by: lizz <lizz@sensetime.com>
2020-03-12 23:20:50 +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 3fa5851891
Update pre-commit hook config (#200)
* update pre-commit hook config

* update the contributing guide
2020-03-01 15:51: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
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
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
Kai Chen 43ca0e57a9
fix the shape assertion in impad (#172) 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
Kai Chen 0eef3757e8
fix the code style of yapf 0.29 (#150)
* fix the code style of yapf 0.29

* fix flake8
2019-12-02 13:14:33 +08:00
Vladimir Iglovikov 56971278db Added Weight and Biases logger. (#143)
* Added wandb logger

* bugfix

* fixes

* fix

* flake8 fixes

* yapf fixes

* Refactor hook, add test

* flake8 fixes

* yapf fixes

* fix in test

* mock package => travis
2019-12-02 12:32:50 +08:00
Kai Chen 8829ff0dfd
Add rgb2gray and gray2rgb (#139)
* add rgb2gray and gray2rgb

* update version

* minor fix for the docstring
2019-10-06 22:46:06 +08:00
Dhananjai Sharma d361ef21b0 Added solarize and posterize image transformations (#132)
* Added solarize and posterize image transformations

* Added solarize and posterize transformations. Modified __init__.py

* Updated mmcv/image/__init__.py

* Updated the docstrings for solarize and posterize

* add unit tests

* fix a test case
2019-10-06 22:12:17 +08:00
lizz 25b422ab6f Let progress goes to stderr (#115)
* let progress goes to stderr

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

* Add argument file=sys.stdout

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

* For the sake of the stupid pytest

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

* line width < 120 is anti-humane

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

* flake8

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

* Lets try

Signed-off-by: lizz <lizz@sensetime.com>
2019-08-21 19:31:58 +08:00
ZwwWayne d5865e0cbd Track iter progress (#112)
* track progress of iter&enum

* reformat

* reformat with yapf

* add unitest

* add doc, and deprecate track_enum_progress

* update docs & comments
2019-08-20 15:05:17 +08:00
Kai Chen eae81c1e86
Add yapf and isort to travis (#96)
* add yapf and isort to travis

* minor formatting

* remove blank lines

* skip unit tests for progressbar when python2

* update travis to ubuntu 16.04

* use a newer version ffmpeg

* add -y to add-apt-repository
2019-07-30 23:15:56 +08:00
Evgeny Nizhibitsky 395efbf2e9 Make latest checkpoint symlink relative (#75)
* Make latest symlink relative. Add tests

* Skip test_save_checkpoint in the absense of torch

* Fix pep8 in test_runner

* Import runner only if torch import has succeeded
2019-06-15 19:56:27 +08:00
nbei 9097957434 C++ implementation for Flow Warp Module (#71)
* flow_warp_c

* flow_warp_c

* beautify format

* beautify format

* beautify format

* beautify format

* add Cython

* modify

* fix details

* fix details

* fix type
2019-05-28 20:03:08 +08:00
Gu Wang d4b91b8376 test bgr2hls 2019-05-07 12:04:01 +08:00
Yue Zhao 944da49fcb add iminvert (#58)
* add iminvert

* fix for flake8

* minor improvement

* add test for iminvert

* fix for flake8

* fix for flake8

* revise __init__

* revise __init__
2019-04-20 00:27:59 -07:00
Kai Chen d426585999
Support str type for loggers (#39)
* fix the time estimation when resuming from a checkpoint

* fix the time estimation when resuming from a checkpoint

* support str type for loggers

* fix yaml dump tests
2019-03-25 21:59:02 -07:00
Kai Chen 47fc7a69ed fix unit test 2018-10-07 19:31:47 +08:00
Kai Chen b98980b77b add some unit tests 2018-10-07 17:45:41 +08:00
Kai Chen 852e980eac ignore a flake8 error 2018-10-04 20:36:29 +08:00
Kai Chen 6e8d925e45 add a class decorator to register file handlers 2018-10-04 19:48:59 +08:00
Kai Chen 51b73ac173 refactoring and remove some public apis 2018-10-04 17:15:14 +08:00
Kai Chen 34eca27216 add register_handler and refactor BaseFileHandler 2018-10-04 16:58:11 +08:00
Kai Chen ddef9a5606 fix flake8 errors 2018-10-01 22:13:52 +08:00
xiongyu 6249f829c6 add negative indexing support 2018-09-27 13:38:36 +08:00
xiongyu c75ff25940 add more tests for video_reader slice 2018-09-27 13:21:59 +08:00
xiongyu c16fab7eca minor change: value error to index error 2018-09-26 22:16:11 +08:00
xiongyu 2ab73c85ee add slice support and unit test 2018-09-26 22:00:30 +08:00
xiongyu 7bea02c0fa update video reader test code 2018-09-21 21:55:37 +08:00
Kai Chen 75ee2531a0 rename *Processor to *Handler and fix bug for python 2 2018-09-21 14:00:56 +08:00
Kai Chen 31e684fcfe bug fix: raise exceptions for missing config keys 2018-09-20 14:15:12 +08:00
Kai Chen 685e8f991b fix unit test for imflip 2018-09-02 21:31:07 +08:00
Kai Chen 1f76784717 add some methods 2018-09-02 21:20:02 +08:00
Kai Chen 978ecfda84 refactoring for optflow 2018-08-28 01:38:53 +08:00
Kai Chen 4adf7006b0 add config class 2018-08-27 20:25:52 +08:00
Kai Chen f872672f58 renaming and reorganizing for subpackage 2018-08-27 17:37:13 +08:00
Kai Chen eadd1e0f4f re-organize io module as a subpackage 2018-08-27 15:29:12 +08:00
Kai Chen 5f2e58c5f2 add optical flow module 2018-08-27 03:10:39 +08:00
Kai Chen 43ebfd9a62 move some unit tests 2018-08-27 02:38:05 +08:00
Kai Chen cbcb90dc21 port from cvbase and do some refactoring 2018-08-27 02:12:31 +08:00