Commit Graph

413 Commits (b20d6ab18b03dfe31515e8c54cb91efa9d82e451)
 

Author SHA1 Message Date
Kai Chen 213156cebb
Allow registering new backends with decorators ()
* 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 2020-05-30 13:09:09 +08:00
Jerry Jiarui XU 5542c24bb1
Fixed package data in pip () 2020-05-30 13:04:37 +08:00
Wenwei Zhang 8ceb404ea6
Use mapping rather than dict for special keys ()
* 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 () 2020-05-30 13:00:01 +08:00
Wenwei Zhang ff89d36612
Add resnext_101_32x8d model ()
* add resnext_101_32x8d model

* rename file

* rm bgr

* Upgrade version
2020-05-27 23:30:16 +08:00
Jerry Jiarui XU d6411b7fff
BC of model zoo: add deprecate urls ()
* 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 ()
* 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 () 2020-05-26 00:39:14 +08:00
Miguel Mendez fa470ebcef
Force copy of input img () 2020-05-24 17:45:21 +08:00
Wenwei Zhang 02f2573a29
Add regnetx links () 2020-05-23 15:34:46 +08:00
Jerry Jiarui XU 0946feabe3
add PetrelBackend ()
* 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 ()" ()
This reverts commit 7f80a2c6b9.
2020-05-23 01:07:43 +08:00
Jerry Jiarui XU 7f80a2c6b9
add PetrelBackend ()
* 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 ()
* add res2net and regnetx

* remove regnetx
2020-05-21 19:05:42 +08:00
Kai Chen 6004776241
fix the install_requires () 2020-05-19 14:41:59 +08:00
Jerry Jiarui XU fabee855b1
Add requirement parser () 2020-05-15 23:08:12 +08:00
Lei Yang eb93599e1b
Add yapf to install_requires () 2020-05-15 20:25:59 +08:00
Kai Chen 27e3284e26
bump the version to 0.5.3 () 2020-05-15 01:51:19 +08:00
Kai Chen feb650642e
Bug fix for the deploy workflow ()
* 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 ()
* 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 ()
* add yapf for pretty text

* add yapf style
2020-05-14 14:17:39 +08:00
wang shiguang a805cd9d91
wrapper extension ()
* 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 ()
* 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 ()
* 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 () 2020-05-12 10:41:11 +08:00
Jerry Jiarui XU d5936d02e3
Make Config.dump() output in original file ()
* add dump() from for different file type

* remove exception
2020-05-11 21:48:56 +08:00
Kai Chen d4fac3a6dd
Use parrots wrapper ()
* 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 () 2020-05-05 21:03:12 +08:00
Kai Chen d2e3a84315
add missing requirements for the doc building () 2020-05-05 14:22:23 +08:00
Kai Chen e54c85e71c
fix the doc building () 2020-05-05 14:07:04 +08:00
Kai Chen 1f6ac2ca85
Add docs for CNN modules ()
* 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 () 2020-05-05 01:04:03 +08:00
Wenwei Zhang 19e4a06cbc
Fix CosineAnealingLr register bug ()
* 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 ()
* 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 ()
* 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 () 2020-05-02 23:57:03 +08:00
Jerry Jiarui XU 1d54ee217e
remove deprecate register_module () 2020-05-02 00:35:30 +08:00
Jerry Jiarui XU a5b5193767
Fixed config import in config file ()
* fixed config import in code

* fixed isort
2020-05-01 21:55:07 +08:00
Kai Chen 45111e193d
Add building bricks of cnn ()
* 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 ()
* Fix Deadlock fix

Resolves 

* 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 ()
* 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 () 2020-04-29 21:30:26 +08:00
Jerry Jiarui XU 45a39dc6a8
make config merge a into b non-inplace () 2020-04-29 15:09:27 +08:00
Kai Chen f7de63fdd0
Support specifying names for registry.register_module() ()
* 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 () 2020-04-27 22:44:26 +08:00
Rui Xu 0c34ab90d0
fix previous bug in save-model ()
* 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 ()
* track progress of iter&enum

* restore

* Add CyclicLrUpdater in Hooks
2020-04-27 13:35:11 +08:00
Kai Chen a338d43d78
Refactor unittests ()
* refactor unittests

* split test_video.py to two files
2020-04-26 22:54:27 +08:00
Cao Yuhang b7e8d7d7fb
Use f-string ()
* 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