mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
594 Commits
8 Branches
41 Tags
32 MiB
Commit Graph

594 Commits (dd660ed99ee4eba797b31fc730ae95e032ef13e6)
 

Author SHA1 Message Date
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
fangxu c090d3f7fd
[Fix] Fix a bug for multiple output in swin transformer. (#571) 
* Update swin_transformer.py

fix bug;

layer_norm 0,1,2,3  will be the same and resulted in an error when out_indices dict exceed one element.

* fix bug, layer_norm 0,1,2,3

* Add unit tests for multiple out_indices

Co-authored-by: cxiang26 <cq.xiang@foxmail.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
 2021-12-03 16:11:41 +08:00
Ezra-Yu 7b37bc5bf5
[Refactor] Refactor `analysis_log.py`. (#529) 
* fix bugs and refactor analysis_log

* update some names
 2021-12-03 16:02:50 +08:00
Ma Zerun d25a78d547
[Enchance] Set a random seed when the user does not set a seed. (#554) 2021-12-02 18:09:55 +08:00
Ma Zerun 78d6d8503f
[Refactor] Use new API of matplotlib to handle blocking input in visualization. (#568) 
* [Refactor] Use new API of matplotlib to handle blocking input in
visualization.

* Modify unit tests
 2021-12-02 17:46:40 +08:00
Ma Zerun 33f049b4c2
[CI] Skip build CI if only configs or docs modification. (#575) 
* [CI] Skip build CI if only configs or docs modification.

* Use ubuntu latest
 2021-12-02 16:05:10 +08:00
takuoko 9d9dce69ad
[Feature] Support using checkpoint in Swin Transformer to save memory. (#557) 
* add checkpoint in swin backbone

* add checkpoint in swin backbone
 2021-12-01 11:49:00 +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
Ma Zerun 00700d570f
[CI] Use CircleCI to do unit tests. (#567) 
* Add .circleci/config.yml

* Use new markdown lint
 2021-11-30 14:22:02 +08:00
mzr1996 6e69f91328 [Docs] Add MMFewshot in README and banner. 2021-11-29 19:00:51 +08:00
juanjompz 715a80b09c
[Enhance] Focal loss for single label tasks (#548) 
* Added focal loss support to single label tasks.

* Updated focal loss docstring
 2021-11-26 12:36:49 +08:00
Ezra-Yu 4b2128fdbd
[Fix] Fix bug when using `ClassBalancedDataset` (#555) 
* fix dataset wrapper bug and add unit tests

* fix dataset wrapper bug and add unit tests

* update unit tests

* fix lint

* align interface to mmdet

* update unit tests
 2021-11-26 09:40:00 +08:00
mzr1996 2266a59a9d Update CI 2021-11-25 15:58:41 +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 f3fbc8b90b
[Docs] Add abstract and image for every paper. (#546) 2021-11-24 17:23:37 +08:00
Ma Zerun ca997c5a79
Update schedule.md 2021-11-22 14:18:01 +08:00
Ma Zerun 23de5cccf9
Update schedule.md 2021-11-22 14:17:26 +08:00
Ma Zerun a55324c453
[Refactor] Remove useless `import_modules_from_string`. (#544) 2021-11-22 14:01:26 +08:00
Ezra-Yu f361bd52e9
[Fix] Fix a bug when using iter-based runner with 'val' workflow (#542) 
* add kwargs and default of optimizer in train_step and val_step

* update docstring

* update docstring

* update optional annotation
 2021-11-22 12:06:47 +08:00
LeoXing1996 49cbfd776a
[Fix] Fix interpolation method checking in `Resize` (#547) 
* fix interpolation method checking in resize

* revise docstring + add unit test
 2021-11-19 17:38:25 +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
takuoko 9f31fc5390
[Refactor] Use `reset_classifier` to remove head of timm backbones. (#534) 
* refactor timm backbone

* fix comment
 2021-11-16 16:24:55 +08:00
xcnick 3eafc5cbe1
[Feature] Add MLP Mixer Backbone. (#528) 
* Add Mlp Mixer

* Fix token_mix FNN add_identity and add out_indices

* Fix mlp_mixer unit test
 2021-11-15 18:25:22 +08:00
huoshuai-dot 3f52488d52
[Docs] Add the top-5 acc in ResNet-CIFAR README. (#531) 
* Update README.md

* Update README.md
 2021-11-12 16:13:41 +08:00
Ma Zerun e0d32df076
[Fix] Fix a bug when load checkpoints in mulit-GPUs environment. (#527) 
* Fix a bug when load checkpoints in mulit-GPUs environment.

* Map the weights to CPU
 2021-11-10 17:54:43 +08:00
Ezra-Yu 87c67a6a79
[Feature] Support positive weights in BCE. (#516) 
* add pos-weight in BCE

* mv pos-weight to kwargs
 2021-11-10 17:14:12 +08:00
Ezra-Yu 34d5a25281
[Refactor] Support passing arguments to loss from head. (#523) 2021-11-10 17:12:34 +08:00
Zhicheng Chen 9ab9d4ff31
[Refactor] Refactor `Resize` transform and add `Pad` transform. (#506) 
* refactor resize, test tobe done

* resize reimpl according to discussion; add pad

* minor fixes and add tests

* minor fixes on docstring

* add additional unit test

* reformat resize and pad

* revise code and docstr according to the comments
 2021-11-10 16:46:49 +08:00
imyhxy dc35eb62b6
[Fix] Fix an error on indexing scalar metrics in `analyze_result.py` (#518) 
* [Fix] Error on indexing scalar metrics

* [Fix] Error on indexing scalar metrics

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

Co-authored-by: Ma Zerun <mzr1996@163.com>
 2021-11-05 11:57:49 +08:00
Ma Zerun 29f512156c
[Docs] Fix TOC of `visualization.md` and add example images. (#513) 2021-11-03 12:21:05 +08:00
imyhxy abb5a222cf
[Fix] Fix wrong condition judgment in `analyze_logs.py` and prevent empty curve. (#510) 
* [Fix] Analyze logs with single training iteration

* [Fix] Make `plot_curve` with 'accuracy' metric

* [Fix] Assert length of training log records of each epoch is larger than 1

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

Co-authored-by: Ma Zerun <mzr1996@163.com>
 2021-11-03 11:37:51 +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 d1473e4a7f
[Dependency] Update mmcv dependency version (#509) 
* Update mmcv dependency version

* Add code info in some metafiles
 2021-11-02 18:08:30 +08:00
mzr1996 b8c3825bb3 Fix uppercase in repvgg metafile 2021-11-02 15:13:26 +08:00
Ma Zerun 72cffac7b5
Bump version to v0.17.0 (#503) 2021-10-29 14:04:03 +08:00
mzr1996 5ac0a1561d [Docs] Fix missing checkpoints links in T2T-ViT. 2021-10-29 10:42:10 +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
Ezra-Yu 2ce5825ef1
[Feature] Support ImageNet21k dataset. (#461) 
* add imagnet21k

* Update unit test

* update imaenet21k

* use slots

* use slots

* rename Data_item to ImageInfo

* add unit tests

* Update unit tests

* rm some print

* update unit tests

* fix lint

* remove default value of pipeline
 2021-10-28 15:22:08 +08:00
imyhxy 671414becb
[Fix] Fix missing import `Compose` and `Normalize`. 
* Fixed missing import 'Compose'

* Fixed mistype `Compose` in `mmcls/datasets/__init__.py`

* Fixed missing import `Normalize`

* [Docs] Fix typos in doctest

* [Fix] Sort import module
 2021-10-28 15:21:05 +08:00
Youqing Xiaozhua 1177feb05b
[Fix] Remove `DistSamplerSeedHook` if use `IterBasedRunner`. (#501) 
* remove DistSamplerSeedHook for IterBasedRunner

* Register DistSamplerSeedHook for EpochBasedRunner only

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

Co-authored-by: Ma Zerun <mzr1996@163.com>
 2021-10-27 17:07:33 +08:00
Ezra-Yu 52e62560d4
[Feature] Add a tool to find broken files. (#482) 
* add verify dataset

* add phase

* rm attr of single_process

* Use `mmcv.track_parallel_progress` to track the validation.

Co-authored-by: mzr1996 <mzr1996@163.com>
 2021-10-27 10:14:19 +08:00
Zhicheng Chen 6d6ce215a2
[Feature] Add seesaw loss. (#500) 
* migrate seesaw loss from mmdet

* add assertion and doc string fixes

* add error information

* docstring fixes

* minor doc update
 2021-10-27 10:11:59 +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