Commit Graph

129 Commits (1717efadbe74842675c586932d0fd566a87495e3)

Author SHA1 Message Date
Ezra-Yu 3482521587
[Feature] Add multi machine dist_train (#734)
* add dist_train with multi machines

* add dist_train with multi machines
2022-03-23 12:03:03 +08:00
Ma Zerun bc781ac3cd
Bump version to v0.21.0 (#722)
* Bump version to v0.21.0

* Update by comment
2022-03-04 16:12:25 +08:00
Ezra-Yu d08c2a148a
[Enhance] Enhance vis-pipeline tool. (#604)
* enhance vis-pipeline add intermediate imgs

* enhance vis-pipeline add intermediate imgs

* improve code of vi-pipeline

* modify docs for vis-pipeline

* Use `mmcv.utils.digit_version` instead of `distutils`

* add size info in the bottom

* preform adaptive-resize in before concat

* add warning info

* fix docs

* fix lint

* fix comment

* fix docs

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-04 14:40:02 +08:00
kusakana 779a06257c
[Feature] Support Wide-ResNet. (#715)
* Add configs of Wide ResNet

* updated config

* add docs and metafile

* update configs

* remove extra import

* Update metafile and readme

Co-authored-by: Ezra-Yu <1105212286@qq.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-04 14:06:34 +08:00
Ma Zerun 9fd35dd7b5
[Reproduction] Reproduce training results of DeiT. (#711)
* Update deit training settings

* Update decay config

* Add mixup&cutmix and drop path rate

* Update training configs

* Update model-zoo

* Add comments
2022-03-02 14:23:10 +08:00
Ma Zerun a7f8e96b31
Bump version to v0.20.1 (#685) 2022-02-07 11:46:50 +08:00
mzr1996 e0edffb353 Merge branch 'dev' 2022-01-31 12:00:17 +08:00
Ma Zerun fb6d6f18d9
Bump version to v0.20.0 (#679)
* Bump version to v0.20.0

* Improve according to comments
2022-01-31 11:51:49 +08:00
Ma Zerun 8ef5aeda40
[Enhance] New-style CPU training and inference. (#674)
* [Enhance] New-style CPU training and inference.

* Add version check in CPU training/test
2022-01-30 20:49:54 +08:00
Ma Zerun 5de480ea9e
[Feature] Support HRNet and add pre-trained models (#660)
* Support HRNet

* Add HRNet configs

* Fix a bug in backward

* Add configs and update docs.

* Not use bias in conv before batch norm

* Defaults to use `norm_eval=False`

* Add unit tests and support out_channels in HRFuseScales

* Update checkpoint path

* Update docstring.

* Remove incorrect files

* Improve according to comments
2022-01-28 10:54:14 +08:00
Ma Zerun dc456a0c2c
[Feature] Support ConvNeXt (#670)
* Support ConvNeXt

* Add configs of ConvNeXt

* Update dev scripts

* Update docs.

* Use new style README

* Add unit tests.

* Update README

* Imporve according to comments

* Modify refers to timm.

* Imporve according to comments
2022-01-28 10:36:45 +08:00
Ezra-Yu 94eb3ff83e
[Feature] Add Twins backbone and convert checkpoints. (#642)
* add twins backbone

* add position_encoding

* refactor twins

* Supplemental unit tests

* update docstring and readme

* update docstring and readme

* update docstring and readme

* update docstring

* update docstring

* update docstring

* update docstring

* remove note

* update doc and docstring

* update docstring

* update docstring

* use abstract pdf link and rename yamlfile

* Update model link

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-01-27 10:25:05 +08:00
Ma Zerun 833152b1f4
[Docs] Update README in configs according to OpenMMLab standard. (#672)
* Update README according to OpenMMLab standard.

* Update model zoo docs generation.

* Revert modification for paperlink
2022-01-26 18:26:01 +08:00
WINDSKY45 b5bd87d7fa
[Docs] Fix some typo in docs. 2022-01-25 12:32:06 +08:00
Zhicheng Chen d56170a734
[Feature] Support EfficientNet (#649)
* add config for resnest test

* fix config

* add label smoothing

* add memcached

* minor fix

* fix bug

* fix config

* add config

* minor fix

* fix configs

* use EResize

* change interpolation

* add more configs

* add docsting

* add unittest

* remove unnecessary changes

* minor fix

* add more docstring

* fix linting

* add efficient backbone

* add config

* add Edge Residual

* fix bug

* remove unnecessary files

* refactor

* add resize in crop to ensure crop size is output size

* fix bug and add comments

* test

* fix

* add more configs

* add more configs

* add more configs

* fix bug

* add model zoo

* fix

* reorganize code

* add edge tpu

* add edge tpu converter

* rename

* update readme

* reorganize code and config

* Rename configs of EfficientNet, and add metafile & model_zoo

* Remove `backend='pillow'`

* Add comments about EfficientNet-EdgeTPU

* Rename the convert tool of EfficientNet.

* Refactor EfficientNet and update docstring.

* Update EfficientNet-EdgeTPU config

* Fix unit tests

Co-authored-by: lixinran <lixr423@outlook.com>
Co-authored-by: lixinran <lixinran@sensetime.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-01-25 12:14:17 +08:00
xiefeifeihu 25f5cc7ea5
[Docs] Fix typo of `analysis.md`. (#664)
test.py not train.py
2022-01-22 10:39:23 +08:00
xiefeifeihu 9e6bf24ec1
[Docs] Fix typo in `analysis.md`. (#663)
示例脚本应执行“eval_metric.py”,而不是analyze_results.py。
另外应该是利用 `tools/test.py` 的 `--out` 参数,而不是`tools/train.py`
2022-01-22 10:38:43 +08:00
Ma Zerun 5232965b17
[Docs] Update install guide and README (#624)
* Update install guide

* Add new repos in README and use readthedocs link preferentially.

* Add MMRazor

* Improve README

* Imporve docs

* Imporve docs
2022-01-04 11:38:05 +08:00
Ma Zerun 7dfc9e4a85
Bump version to v0.19.0 (#623) 2021-12-31 12:55:47 +08:00
Zhicheng Chen 0bbbb04429
[Enhancement] Improve the accuracy of ResNet (#572)
* add itp timm

* minor update

* minor update

* minor update

* add rep aug, minor update on configs

* minor update

* add target threshold

* add decaymulti

* minor update

* minor update

* add lbl smooth

* update lr

* reorganize config files and code

* minor bugfixes

* remove unused parts and minor fixes on cfg

* critical bugfix, add test and cfg update

* refactor code

* update doc string

* remove duplicate code

* refactor drop path in resnet

* rename

* Modify configs and add README&metafile

* Update metafile

Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-29 10:58:42 +08:00
Ezra-Yu e7c06b8541
[Reproduction] Reproduce RegNetX training accuracy. (#587)
* add preciseBN hook

* add precise-bn config

* add proority of preciseBN

* Add precise BN hook

* Update PreciseBN hook

* add config example & pre-commit

* Update PBN

* Add unit test

* Update unit tests

* mv preciseBN to mmcls.runner

* rm extra sleep()

* merge latest

* update code

* fix lint

* update unit tests

* add regnetx configs

* update prcise BN in configs

* fix lint

* add checkpoints and logs

* fix conflict

* fix lint

* update model_zoo and model-index

* remove some changes

* update readme.md

* remove precise_bn.py
2021-12-28 17:34:27 +08:00
Ma Zerun 159b38d276
[Reproduction] Reproduce training results of T2T-ViT (#610)
* Add cosine cool down lr updater

* Use ema hook

* Update decay mult

* Update configs.

* Update T2T-ViT readme and format all readme

* Update swin readme

* Update tnt readme

* Add docstring for `CosineAnnealingCooldownLrUpdaterHook`.

* Update t2t readme and metafile
2021-12-28 15:09:40 +08:00
Ezra-Yu 131d8c6296
[Feature] Add CAM visualization tool (#577)
* add cam-grad tool

* refactor cam-grad tool

* add docs

* update docs

* Update docs and support Transformer

* remove pictures and use link

* replace example img and finish EN docs

* improve docs

* improve code

* Fix MobileNet V3 configs

* Refactor to support more powerful feature extraction.

* Add unit tests

* Fix unit test

* fix distortion of visualization exapmles in docs

* fix distortion

* fix distortion

* fix distortion

* merge master

* merge fix conficts

* Imporve the tool

* Support use both attribute name and index to get layer

* add default get_target-layers

* add default get_target-layers

* update docs

* update docs

* add additional printt info when not using target-layers

* Imporve docs

* Fix enumerate list.

Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-23 18:53:40 +08:00
mzr1996 b4450b720e [Docs] Fix colab link 2021-12-17 18:07:19 +08:00
Ma Zerun 643fb192cd
[Enhance] Enhance feature extraction function. (#593)
* Fix MobileNet V3 configs

* Refactor to support more powerful feature extraction.

* Add unit tests

* Fix unit test

* Imporve according to comments

* Update checkpoints path

* Fix unit tests

* Add docstring of `simple_test`

* Add docstring of `extract_feat`

* Update model zoo
2021-12-17 15:55:02 +08:00
Ma Zerun f9a2b04cee
[Feature] Add DeiT backbone and checkpoints. (#576)
* Support DeiT backbone.

* Use hook to automatically resize pos embed

* Update ViT training setting

* Add deit configs and update docs

* Fix vit arch assertion

* Remove useless init function

* Add unit tests.

* Fix resize_pos_embed for DeiT

* Improve according to comments.
2021-12-15 22:44:57 +08:00
Ma Zerun 6f25bebe42
[Docs] Update docs. (#600)
* [Docs] The template provides united Docs and OpenMMLab drop-lists in
banner. Remove them from our `conf.py`.

* Move docs of different languages to the same folder
2021-12-15 10:34:34 +08:00
mzr1996 e57b8cb33b [Fix] Fix docs generation. 2021-12-14 16:23:56 +08:00
Ezra-Yu 72b0da8bd7
[Docs] Update mmcv, torch, cuda version in Dockerfile and docs (#594) 2021-12-08 18:59:21 +08:00
Zhiliang Peng 18f6bb0b10
[Feature] Implement the conformer backbone. (#494)
* implement the conformer

* format code style

* format code style

* reuse the TransformerEncoderLayer in the vision_transformer.py

* Modify variable name

* delete unused params

* Remove warning info in Conformer head since it already exists in
Conformer.

* Rename some variables

* Add unit tests

* Use `getattr` instead of `get_submodule`.

* Remove some useless layers

* Refactor conformer and add configs

* Update configs and add metafile.

* Fix unit tests

* Update README

Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-07 14:00:17 +08:00
Ezra-Yu fdb178303b
[Docs] Add analysis&misc docs (#525)
* refactor ananlysis_log and add docs

* refactor ananlysis_log and fix lint

* improve docs

* improve docs

* improve docs

* fix bugs and refactor analysis_log

* rename analysis folder to analysis_tools

* fix failure link

* add result analysis docs

* add eval-metrics docs

* add misc doc

* fix lint

* improve docs

* improve misc docs

* fix docs

* Change the `eval_options` in `tools/analysis_tools/eval_metric.py` to
`metric_options`

* Improve tutorials

Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-07 11:27:34 +08:00
Ezra-Yu b3a6cae522
[Docs] Fix failure links (#586)
* Update T2T link

* Add ResNeSt comfig links
2021-12-07 11:26:36 +08:00
Ma Zerun f2c1c57398
[Docs] Fix docs build dependency. (#584)
* Fix docs.

* Add MMHuman3D
2021-12-07 11:16:33 +08:00
Ma Zerun f6076bfeca
Bump version to v0.18.0 (#569)
* Bump version to v0.18.0

* Add highlight of previous version in readme

* Imporve docs
2021-11-30 19:04:38 +08:00
Ezra-Yu 4ddfd3d508
[Docs] Fix vit config and model broken links (#564)
* Update model_zoo.md

* Update model_zoo.md

* fix config link

* fix config link
2021-11-30 16:49:15 +08:00
mzr1996 6e69f91328 [Docs] Add MMFewshot in README and banner. 2021-11-29 19:00:51 +08:00
Zhicheng Chen fc8adbc149
[Enhance] Add metafile, readme and converted models for MLP-Mixer (#539)
* add pth converter

* minor update on config files, add metafile and readme

* add missing readme and minor fixes

* minor fixes

* Update config names and checkpoint download link

* Update model_zoo.md

Co-authored-by: mzr1996 <mzr1996@163.com>
2021-11-24 19:04:19 +08:00
Ma Zerun 23de5cccf9
Update schedule.md 2021-11-22 14:17:26 +08:00
Ma Zerun 7977dc8e2d
[Improvement] Rename config files according to the config name standard. (#508)
* Update tnt config

* Rename config files

* Update docs

* Update metafile in dev_scripts

* Fix some files

* Remove fp16 metafile and README.

* Fix names in metafiles.
2021-11-19 14:20:35 +08:00
Ma Zerun 598ecd4008
[Docs] Add mmflow and mim in banner and readme (#543) 2021-11-19 11:33:17 +08:00
Ezra-Yu 771d10548c
[Docs] Add schedule and runtime tutorial docs (#499)
* add cn tutorials/config.md

* add heads api and doc title link

* Update tutorials index

* Update tutorials index

* Update config.md

* add english version

* Update config.md

* add custom_runtime

* Update docs

* modify title

* modify en to zh_CN in chinses docs

* Update Readme

* fix punctuations

* Update docs/tutorials/customize_runtime.md

Co-authored-by: Ma Zerun <mzr1996@163.com>

* Update docs/tutorials/customize_runtime.md

Co-authored-by: Ma Zerun <mzr1996@163.com>

* split to schedule and runtime

* fix lint

* improve docs after review

* fix TOC

* imporve expersion

* fix an error

* Imporve schedule.md

* Improve runtime.md

* Improve chinese docs.

* Fix toc-tree

* fix en link and add a case of gradient clipping

* fix wrong word

Co-authored-by: Ma Zerun <mzr1996@163.com>
2021-11-17 18:12:54 +08:00
Ma Zerun 29f512156c
[Docs] Fix TOC of `visualization.md` and add example images. (#513) 2021-11-03 12:21:05 +08:00
Ezra-Yu b486bb23f8
[Feature] Add a tool to visualize learning rate in each iterations (#498)
* add vis lr

* add doc

* Update doc and rm load_json_log in analysis

* Update docs

* add unit tests and update docstring

* fix unit-tests

* Use DummyIterBasedRunner and DummyEpochBasedRunner to simulate training
process.

* rm function val and add judge

* update docs

* improve english expression

* Update docs/tools/visualization.md

Co-authored-by: Ma Zerun <mzr1996@163.com>

* fix typo and ImageNet datasize

* fix typo

* update example

Co-authored-by: mzr1996 <mzr1996@163.com>
2021-11-03 10:48:56 +08:00
Ma Zerun 82713c9ba4
[Docs] Use docs link of other projects and add MMCV docs. (#511)
* [Docs] Use docs link of other projects and add MMCV docs.

* Fix misused "document", and it should be "documentation"

* Fix uppercase

* Rename `Projects` to `Docs`.
2021-11-03 10:42:45 +08:00
Ma Zerun 72cffac7b5
Bump version to v0.17.0 (#503) 2021-10-29 14:04:03 +08:00
Ma Zerun fffa30dd48
[Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467)
* add t2t backbone

* register t2t_vit

* add t2t_vit config

* [Temp] Align posterize transform with timm.

* Fix lint

* Refactor t2t-vit

* Add config for t2t-vit

* Add metafile and README for t2t-vit

* Add unit tests

* configs

* Update metafile and README

* Improve docstring

* Fix batch size which should be 8x64 instead of 8x128

* Fix typo

* Update model zoo

* Update training augments config.

* Move some arguments of T2TModule to T2TViT

* Update docs.

* Update unit test

Co-authored-by: HIT-cwh <2892770585@qq.com>
2021-10-29 10:37:16 +08:00
Ma Zerun 54ef176235
[Docs] Add English Colab tutorials. (#497)
* Add English-version Colab tutorials

* Revise Chinese-version Colab tutorials.

* Add colab tutorials menu on readthedocs page.

* Fix lint
2021-10-26 17:40:24 +08:00
Ezra-Yu 3a35b6f956
[Docs] Add tutuorial for config. (#487)
* add cn tutorials/config.md

* add heads api and doc title link

* Update tutorials index

* Update tutorials index

* Update config.md

* add english version

* Update config.md

* Update docs

* Update css

* Update docs/tutorials/config.md

Co-authored-by: Ma Zerun <mzr1996@163.com>

* Update docs_zh-CN/tutorials/config.md

Co-authored-by: Ma Zerun <mzr1996@163.com>

* modify according to suggestion

* Use GitHub style `code` css

* change some mmcv API link to CN version

* remove default in default_runtime

Co-authored-by: mzr1996 <mzr1996@163.com>
2021-10-26 16:43:33 +08:00
Ma Zerun 77a3834531
[Feature] Add Res2Net backbone and converted weights. (#465)
* Add Res2Net from mmdet, and change it to mmcls style.

* Align structure with official repo

* Support `deep_stem` and `avg_down` option

* Add Res2Net configs

* Add metafile&README and update model zoo

* Add unit tests

* Imporve docstring.

* Improve according to comments.
2021-10-20 16:34:22 +08:00
Ezra-Yu 9dbe58bf8e
[Feature] Add pipeline visualization tools. (#406)
* add vis

* add tool vis-pipeline

* add docs

* Update docs

* pre-commit

* enhence english expression

* Add `BaseImshowContextmanager` and `ImshowInfosContextManager` to reuse
matplotlib figure.

* Use context manager to implement `imshow_infos`

* Add unit tests.

* More general base context manager.

* unit tests for context manager.

* Improve docstring.

* Fix context manager exit cannot close figure when matplotlib>=3.4.0

* Fix unit tests

* fix lint

* fix lint

* add adaptive

* add adaptive

* update adaptive

* add GAP

* improve doc and docstring

* add visualization in doc index

* Update doc

* Update doc

* Update doc

* Update doc

* Update doc

* Update doc

* update docs and docstring

* add progressbar

* add progressbar

* add images

* add images

* Delete .DS_Store

* replace images

* replace images and modify rgb2bgr

* add picture size

* mv pictures

* update img display

* add doc_zh-CN images

* Update vis_pipeline.py

* Update visualization.md

* Update visualization.md

* fix lint

* Improve docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
2021-10-20 10:28:21 +08:00