Commit Graph

289 Commits (357b0dfb366e5d088de08e8fbe914f2a0f19f3b1)

Author SHA1 Message Date
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