Commit Graph

299 Commits (6ce22a13e69b1c99afe0234f3e678a83d8afbcd2)
 

Author SHA1 Message Date
Jerry Jiarui XU 6ce22a13e6
Add res2net pretrain model (#289)
* add res2net and regnetx

* remove regnetx
2020-05-21 19:05:42 +08:00
Kai Chen 6004776241
fix the install_requires (#288) 2020-05-19 14:41:59 +08:00
Jerry Jiarui XU fabee855b1
Add requirement parser (#286) 2020-05-15 23:08:12 +08:00
Lei Yang eb93599e1b
Add yapf to install_requires (#285) 2020-05-15 20:25:59 +08:00
Kai Chen 27e3284e26
bump the version to 0.5.3 (#284) 2020-05-15 01:51:19 +08:00
Kai Chen feb650642e
Bug fix for the deploy workflow (#283)
* update the deploy workflow

* fix a typo

* update MANIFEST.in to include pyx

* remove debug info
2020-05-14 23:00:22 +08:00
Kai Chen 506df8cdce
Add a pypi workflow and bump version to 0.5.2 (#282)
* add a workflow to publish to pypi and remove travis.yml

* move the if-statement to the job level

* add a badge

* switch to ubuntu-latest

* bump version to 0.5.2

* add apt-get update before installing
2020-05-14 16:41:40 +08:00
Jerry Jiarui XU 83bcd1134d
Add yapf for Config.pretty_text (#281)
* add yapf for pretty text

* add yapf style
2020-05-14 14:17:39 +08:00
wang shiguang a805cd9d91
wrapper extension (#279)
* wrapper extension

* fix

Co-authored-by: wangshiguang <wangshiguang@sensetime.com>
2020-05-14 14:15:58 +08:00
Kai Chen 693f61af76
Add GitHub action flow (#280)
* Create a github action workflow

* Fix the workflow config

* Fix codecov

* skip codecov for python 3.8

* fix the expression

* remove the if expression
2020-05-14 10:26:27 +08:00
Jintao Lin 25d07e3d54
Support conv layers' own `init_weights` method (#278)
* Support conv layers' own `init_weights` method

* Add related unittest about ConvModule init_weight

* make the comments more specific
2020-05-13 22:58:16 +08:00
Wenwei Zhang c35c228cbb
Support to pass args to cnn bricks (#277) 2020-05-12 10:41:11 +08:00
Jerry Jiarui XU d5936d02e3
Make Config.dump() output in original file (#275)
* add dump() from for different file type

* remove exception
2020-05-11 21:48:56 +08:00
Kai Chen d4fac3a6dd
Use parrots wrapper (#272)
* use parrots wrapper for norms

* add unittests for is_norm()

* add a test case
2020-05-08 11:24:35 +08:00
Kai Chen 6aa5131559
move test_weight_init to tests/test_cnn (#270) 2020-05-05 21:03:12 +08:00
Kai Chen d2e3a84315
add missing requirements for the doc building (#269) 2020-05-05 14:22:23 +08:00
Kai Chen e54c85e71c
fix the doc building (#268) 2020-05-05 14:07:04 +08:00
Kai Chen 1f6ac2ca85
Add docs for CNN modules (#266)
* add docs for cnn modules

* minor fix

* update format

* minor fix

* bump version to v0.5.1
2020-05-05 12:02:33 +08:00
Jerry Jiarui XU 8441f2310b
Update config Error msg (#267) 2020-05-05 01:04:03 +08:00
Wenwei Zhang 19e4a06cbc
Fix CosineAnealingLr register bug (#265)
* Fall back to CosineLr

* Fix consineanealing with unittest

* Cover momentum hook

* Add comments to explain
2020-05-04 00:38:55 +08:00
Xintao 65fbc75689
Add colorspace functions: rgb2ycbcr, bgr2ycbcr, ycbcr2rgb and ycbcr2bgr (#259)
* update

* add functions

* update tests

* resolve comments

* resolve comments

* resolve comments

* update docstr
2020-05-03 23:32:48 +08:00
wang shiguang 5dba1c00bd
Parrots wapper (#258)
* add parrots wapper

* Update parrots_wrapper.py

* Update parrots_wrapper.py

* fix commits

* fix

* fix lint and adapt torch.__version__

* mirror fix

Co-authored-by: wangshiguang <wangshiguang@sensetime.com>
2020-05-03 22:21:27 +08:00
Kai Chen 44e6b93df4
bump version to v0.5.0 (#263) 2020-05-02 23:57:03 +08:00
Jerry Jiarui XU 1d54ee217e
remove deprecate register_module (#262) 2020-05-02 00:35:30 +08:00
Jerry Jiarui XU a5b5193767
Fixed config import in config file (#261)
* fixed config import in code

* fixed isort
2020-05-01 21:55:07 +08:00
Kai Chen 45111e193d
Add building bricks of cnn (#247)
* add building bricks of cnn

* add unit tests

* use registry for building bricks

* minor updates

* add scale layer

* add test for scale

* add doc string

Co-authored-by: Jiarui XU <xvjiarui0826@gmail.com>
2020-05-01 00:32:25 +08:00
Maxim Bonnaerens 89f709e8e7
Fix Deadlock fix (#256)
* Fix Deadlock fix

Resolves #255

* Revert deadlock fix and fix of deadlock fix

This reverts commit af85591fc7.
2020-04-29 23:38:19 +08:00
Rui Xu d27f1f05a8
add exp name in text log (#244)
* add exp name in text log

* fix bugs

* test

* fix yapf bug

* fix meta == None

* fix f-string bug in py3.5

* fix comments from yuhang: do not print exp name every log

* use default implementation to print log

* use f-string

* remove \t and update docstring

* update docstring

Co-authored-by: xurui <xurui4@sensetime.com>
2020-04-29 22:16:42 +08:00
Maxim Bonnaerens f97a8b37ac
Fix fstring in TextLoggerHook (#257) 2020-04-29 21:30:26 +08:00
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