Commit Graph

663 Commits (b5e1facc85219770b6e85a9cfb2ec554167ccedc)
 

Author SHA1 Message Date
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
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
2019-12-12 22:18:40 +08:00
Kai Chen 2330c4f468
Add pre-commit hooks (#156)
* add pre-commit hooks

* update yapf version
2019-12-12 21:40:42 +08:00
Kai Chen fd3f862edb
add an argument to skip os.symlink (#155) 2019-12-12 20:41:43 +08:00
Zhipeng Han 50d2cbdd58 process the case that no weight in the module (#153)
Signed-off-by: han_hans <han_hans@apple.com>
2019-12-12 19:16:16 +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
Rinat Shigapov aea750051f fix dependencies (#146)
* fix requirements

* don't reuse build dependencies between different envs

* minor fix

* change the order of opencv and opencv-headless
2019-11-21 23:34:26 +08:00
Korabelnikov Aleks b480ee475f prevent from hangning in cv2.imshow()/cv2.waitkey (#145)
* prevent from hangning in cv2.imshow()/cv2.waitkey

case:
Window has been closed after cv2.imshow; cv2.waitKey;
expected behavior:
- continue program
actual  behavior:
- program hangs

This commit fixes it

* fix linter
2019-11-17 17:16:45 +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
Kai Chen 688e6abe88
call dist.is_initialized() after checking the availability (#128) 2019-09-07 11:40:33 +08:00
Zhipeng Han 22ba143d6c [feat]: find root of Version Control Systems (#129)
Signed-off-by: Zhipeng Han <hanzhipeng9@gmail.com>
2019-09-07 11:14:32 +08:00
Kai Chen 5b10dcd79f
Update version to v0.2.13 (#117) 2019-08-22 19:59:58 +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 9de7927a69
Use torch.utils.tensorboard instead of tensorboardX (#108)
* use torch.utils.tensorboard instead of tensorboardX

* add backward compatibility

* add .idea to gitignore
2019-08-11 13:30:20 +08:00
Jerry Jiarui XU 27ce41907f fixed crop of single channel image like semantic map (#107) 2019-08-11 12:53:23 +08:00
ZongYuan Zhan 3b46ddf0d3 Fix spelling mistake in Config.auto_argparser. (#106) 2019-08-09 00:01:27 +08:00
Kai Chen 7cb4d804e2
update to v0.2.12 (#104) 2019-08-06 20:57:50 +08:00
Yang Li 067d5086f0 Fix multiple err_msg logging in distribuited mode (#103) 2019-08-06 20:42:43 +08:00
CoinCheung c6f2caea5d resubmit parameter shape check (#100)
* resubmit parameter shape check

* check lint

* print a table of mismatched keys
2019-08-02 16:34:20 +08:00
Kai Chen d1abbca1d9
Create CONTRIBUTING.md (#102) 2019-07-31 20:40:59 +08:00
Karlind 95b87687ca install cython automatically as a dependency (#101)
* install cython automatically as a dependency

* install numpy automatically before building

* fix linting errors
2019-07-31 20:29:04 +08:00
Kai Chen 5f8fb7d463
update version to v0.2.11 (#98) 2019-07-31 02:33:58 +08:00
Kai Chen 5b72bd4baf
fix the compiling on mac os (#97) 2019-07-31 02:22:20 +08:00
Karlind 30985b2f76 add support for pathlib to load and dump (#91)
* add support for pathlib to load and dump

* minor formatting
2019-07-31 01:21:05 +08:00
HarryHan 5a5c94b68b Update lr_updater.py (#95)
* Update lr_updater.py

"Add min_lr to PolyLrUpdaterHook, so that it can anneal to min_lr. When min_lr=0, it behaves as usual. "

* Update lr_updater.py
2019-07-30 23:23:54 +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
Kai Chen 3e1e297d36
use torchvision:// instead of modelzoo:// (#89) 2019-07-13 13:43:10 +08:00
Kai Chen e5ca8846c5
Fix the compiling bug when using `pip install mmcv` (#88)
* rename flowWarp to FlowWarp

* fix the compiling bug when using pip install

* update version to v0.2.10
2019-07-07 21:41:31 +08:00
Kai Chen 0ad15a508e
Add mmcv._ext to autodoc_mock_imports (#86) 2019-07-07 17:32:40 +08:00
Kai Chen e60c7efde3
Improve the documentation (#85)
* fix doc generation

* add docs for optical flow

* fix typo
2019-07-07 16:54:22 +08:00
simon wu 492228794b Fixing build-errors on Windows (#84) 2019-07-07 13:55:51 +08:00
Jiangmiao Pang 258d4d320c fix one bug in vgg (#83) 2019-06-24 20:13:33 +08:00
Kai Chen ff5631003f
add mmaction to readme (#82) 2019-06-24 20:11:30 +08:00
Yue Zhao f9a750fe64 add BN pretrain model, add i3d pretrained on K-400 (#81)
* add BN pretrain model, add i3d pretrained on K-400

* fix flake8
2019-06-18 01:40:11 +08:00
lizz 28feffdb22 Ignore autogenerated flow_warp_module.cpp (#80)
Signed-off-by: lizz <363664470@qq.com>
2019-06-15 22:13:27 +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
lizz 123720b64e Download one copy in distributed environment (#76)
* Download one copy in distributed environment

Signed-off-by: innerlee <363664470@qq.com>

* fix

Signed-off-by: innerlee <363664470@qq.com>

* format

Signed-off-by: innerlee <363664470@qq.com>
2019-06-15 14:52:53 +08:00
Yue Zhao 92a81b62a6 Polish DataContainer (#79)
* add iminvert

* DC now supports dimension other than 3

* fit flake8

* generalize pad_dim and beautify code

* fix bug; rename "pad_dim" to "pad_dims" for disambiguity

* fix bug

* fix flake8

* fix yapf format

* use deafault yapf

* beautify style
2019-06-15 01:00:28 +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
Kai Chen 20d7cd30dc
Merge pull request #70 from hellock/master
Update to v0.2.8
2019-05-22 13:41:42 +08:00
Kai Chen 2051f4f4e6 update to v0.2.8 2019-05-22 13:35:52 +08:00
Kai Chen c234cdea80
add model urls for HRNet (#68) 2019-05-22 11:21:33 +08:00
Cao Yuhang 8e84059cee only display base lr when there are multi lrs (#65)
* only display base lr when there are multi lrs

* change comment: base lr to lr of the first param group
2019-05-13 20:56:42 -07:00