317 Commits

Author SHA1 Message Date
Kai Chen
35ba152821
Add a BaseRunner and rename Runner to EpochBasedRunner (#290)
* add a BaseRunner and rename Runner to EpochBasedRunner

* fix the train/val step

* bug fix

* update unit tests

* fix unit tests

* raise an error if both batch_processor and train_step are set

* add a unit test
2020-06-02 22:23:21 +08:00
Harry
6f21d8b560
Add optimizer constructor from mmdetection (#313)
* feat: add optimizer constructor

* refactor: version
v0.5.9
2020-06-02 19:57:24 +08:00
Miguel Mendez
fe4f657f03
Fix deprecation warning collections.abc (#312)
* Fix deprecation warning collections.abc

* Fix isort
2020-06-02 19:29:54 +08:00
Harry
efecf7d1fe
fix building ConvModule with Tanh activation (#309)
* fix: tanh has not inplace

* feat: add ConvModule with Tanh
2020-06-01 21:26:51 +08:00
Kai Chen
213156cebb
Allow registering new backends with decorators (#307)
* allow registering new backends with decorators

* add a docstring

* minor update to the docstring
2020-05-31 21:56:03 +08:00
Kai Chen
4150d0ac2f
Update build.yml v0.5.8 2020-05-30 13:09:09 +08:00
Jerry Jiarui XU
5542c24bb1
Fixed package data in pip (#303) 2020-05-30 13:04:37 +08:00
Wenwei Zhang
8ceb404ea6
Use mapping rather than dict for special keys (#304)
* Support path as a key in dict of config

* reformat test case

* update pre-commit version and fix format

* fix bug

* clean code

* reformat

* fix missing parts
2020-05-30 13:03:33 +08:00
Wang Xinjiang
b63d880af5
Add sleep(2) during epoch transition (#306) 2020-05-30 13:00:01 +08:00
Wenwei Zhang
ff89d36612
Add resnext_101_32x8d model (#300)
* add resnext_101_32x8d model

* rename file

* rm bgr

* Upgrade version
v0.5.7
2020-05-27 23:30:16 +08:00
Jerry Jiarui XU
d6411b7fff
BC of model zoo: add deprecate urls (#301)
* add deprecate urls

* add deprecate urls

* warning test

* rename to deprecated.json
2020-05-27 22:09:06 +08:00
Jerry Jiarui XU
37d8facfad Support default/external json for open-mmlab models (#230)
* support default/external json for open-mmlab models

* add local

* add more test

* add docs

* add docs

* update docs

* refactor

* add json in MANIFEST

* fixed json typo
2020-05-27 17:12:43 +08:00
Kai Chen
c63e4d57c3
bump version to 0.5.6 (#298) v0.5.6 2020-05-26 00:39:14 +08:00
Miguel Mendez
fa470ebcef
Force copy of input img (#297) 2020-05-24 17:45:21 +08:00
Wenwei Zhang
02f2573a29
Add regnetx links (#287) 2020-05-23 15:34:46 +08:00
Jerry Jiarui XU
0946feabe3
add PetrelBackend (#294)
* add PetrelBackend

* update docs

* add path_maps for Ceph

* rename to path_mapping

* fixed import
2020-05-23 01:31:11 +08:00
Kai Chen
3bf3e8ef24
Revert "add PetrelBackend (#291)" (#293)
This reverts commit 7f80a2c6b9c7f994b9420e72b32f84a0283e4e05.
2020-05-23 01:07:43 +08:00
Jerry Jiarui XU
7f80a2c6b9
add PetrelBackend (#291)
* add PetrelBackend

* update docs

* add path_maps for Ceph

* rename to path_mapping
2020-05-22 12:31:34 +08:00
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) v0.5.5 2020-05-19 14:41:59 +08:00
Jerry Jiarui XU
fabee855b1
Add requirement parser (#286) v0.5.4 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) v0.5.3 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
v0.5.2
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
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) v0.5.0 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 af85591fc7f226623bc9ebef93b76653bfac3438.
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