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

638 Commits (12d8c78222995f5cdd29e67a8ffca4b0194b3e5c)
 

Author SHA1 Message Date
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
Ma Zerun 08bf04e5ee
[Fix] Fix bug in `1-benchmark_valid.py` if checkpoint_root is None. (#489) 2021-10-25 12:01:03 +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
takuoko f68f17e9bb
[Fix] Set the priority of `EvalHook` to "LOW" to avoid a bug of `IterBasedRunner` (#488) 
* Set the priority of EvalHook to LOW.

* add a comment
 2021-10-20 16:24:52 +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
Ma Zerun 2932f9d8a3
[Refactor] Refator ViT (Continue #295) (#395) 
* [Squash] Refator ViT (from #295)

* Use base variable to simplify auto_aug setting

* Use common PatchEmbed, remove HybridEmbed and refactor ViT init
structure.

* Add `output_cls_token` option and change the output format of ViT and
input format of ViT head.

* Update unit tests and add test for `output_cls_token`.

* Support out_indices.

* Standardize config files

* Support resize position embedding.

* Add readme file of vit

* Rename config file

* Improve docs about ViT.

* Update docstring

* Use local version `MultiheadAttention` instead of mmcv version.

* Fix MultiheadAttention

* Support `qk_scale` argument in `MultiheadAttention`

* Improve docs and change `layer_cfg` to `layer_cfgs` and support
sequence.

* Use init_cfg to init Linear layer in VisionTransformerHead

* update metafile

* Update checkpoints and configs

* Imporve docstring.

* Update README

* Revert GAP modification.
 2021-10-18 16:07:00 +08:00
Ezra-Yu 0381309921
[Docs] Update Chinese Colab tutorials. (#483) 
* fix *.ipynb

* fix *.ipynb

* fix lint
 2021-10-15 17:37:12 +08:00