Commit Graph

132 Commits (14dcb69092b4847d968cf8ec1423ff6c004f06b6)

Author SHA1 Message Date
Wangbo Zhao(黑色枷锁) 14dcb69092
[Feature] Add mixmim backbone with checkpoints. (#1224)
* add mixmim backbone

* add mixmim inference

* add docstring, metafile, test and modify readme

* Update README and metafile

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-20 16:52:54 +08:00
Songyang Zhang 5547f4cac4
[Feature] Add TinyViT for dev-1.x. (#1042)
* [Feature] add TinyViT for dev-1.x

* [Feature] update readme

* fix lint error

* refactor the code

* [Update] update the args

* [Update] add unit test and fix bugs

* Rename the configuration file

* delete invalid files

* [Feature] update tinyvit readme

* [Feature] update tinyvit readme

* [Feature] update metafile

* Update tinyvit metafile
2022-12-20 13:04:00 +08:00
takuoko 9e82db6032
[Enhance] Support ConvNeXt More Weights. (#1240)
* convnext more weights

* Update metafile and README

* Fix link

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-19 17:17:57 +08:00
Ma Zerun 3006fa26ab
[Fix] Fix CAM visualization. (#1248) 2022-12-19 13:54:52 +08:00
takuoko 2535c1ecd7
[Feature] Support EVA. (#1239)
* add eva

* add eva

* add eva

* sklearn -> scikit-learn

* add large

* Update model names and links.

* Fix resize pos embed error when loading fp16 weight.

* Remove verbose configs.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-14 13:21:33 +08:00
Songyang Zhang 210373c093
[Feature] Implementation of RevViT. (#1127)
* [Feature] implement rev-vit network

* can reproduce the RevViT-Small accuracy 79.9

* update

* [Feature] update revvit

* [Feature] update revvit readme

* Update links

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-14 11:46:39 +08:00
kitecats ef3610d962
[Bug] Fix `reparameterize_model.py` doesn't save meta info. (#1221)
* fix reparameterize_model.py don't save meta info

* fix error

* Fix symbol link

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-06 17:06:31 +08:00
takuoko 4fb44f8770
[Feature] EfficientNets NoisyStudent & L2. (#1122)
* add mvit 21k

* add mvit 21k

* add effnet noisy student

* Revert "add mvit 21k"

This reverts commit f51067c559.

* revert mvit pr

* update link and readme

* update readme

* update l2 link

* update link

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-11-21 16:57:13 +08:00
marouane amzil 743ca2d602
[Fix] Fix the torchserve. (#1143)
* rebase

* update docker and rm deprecated deployment tools

* update docs

* rebase

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-11-21 11:18:42 +08:00
Ezra-Yu 4969830c8a
[Enhance] Reproduce mobileone training accuracy. (#1191)
* add switch hook and UTs

* update doc

* update doc

* fix lint

* fix ci

* fix ci

* fix typo

* fix ci

* update configs names

* update configs

* update configs

* update links

* update readme

* update vis_scheduler

* update metafile

* update configs

* rebase

* fix ci

* rebase
2022-11-21 10:43:34 +08:00
Jiahao Wang 72c6bc4864
[Feature] Support RepLKnet backbone. (#1129)
* update replknet configs

* update replknet test

* update replknet model

* update replknet model

* update replknet model

* update replknet model

* Fix docs and config names

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-21 10:18:58 +08:00
Songyang Zhang f458bf5a64
[Docs] update visualization doc. (#1160)
* [Docs] update visualization doc

* update doc

* update folder

* update analysis

* Update print config tool

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-18 12:44:55 +08:00
takuoko c4f3883a22
[Feature] Support DaViT. (#1105)
* add davit

* fix mixup config

* convert scripts

* lint

* test

* test

* Add checkpoint links.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-16 17:23:55 +08:00
Austin Welch ee9ee9cf3c
[Fix] Update mmcls2torchserve. (#1189) 2022-11-14 14:17:11 +08:00
Ma Zerun c48cfa9f47
[Docs] Improve the ResNet model page. (#1118)
* [Docs] Improve the ResNet model page.

* Fix lint.

* Update stat.py
2022-11-08 11:09:51 +08:00
takuoko d05cbbcf9b
[Feature] Support HorNet Backbone for dev1.x. (#1094)
* add hornet

* add hornet

* fix mixup config

* add optim cfgs

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-11-04 15:33:46 +08:00
Ma Zerun 29c46c8af2
[Docs] Add runtime configuration docs. (#1128)
* [Docs] Add runtime configuration docs.

* Fix grammar errors.

* Imporve docs according to comments
2022-11-02 10:59:59 +08:00
takuoko 9bc58745d1
[Enhance] Update analyze_results.py for dev-1.x. (#1071)
* update analyze_results

* lint

* add --rescale-factor and fix filename logic

* lint
2022-10-11 11:34:23 +08:00
takuoko a49c3076e1
[Feature] Support DeiT3. (#1065)
* deit3

deit3

lint

* add tools and test

* deit3

* deit3

* fix preprocess

* lint

* Update config names and checkpoint paths

* Update convert tools to use mmengine, and fix docstring.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-10-10 14:54:20 +08:00
Ezra-Yu 080eb79f94
[Fix] Fix verify dataset tool in 1.x (#1062) 2022-09-30 10:08:02 +08:00
Songyang Zhang 81f10ccbb7
[Feature] Support EdgeNeXt for dev-1.x. (#1037)
* [Feature] add EdgeNeXt for dev-1.x

* [Feature] add update

* Imporve EdgeNeXt

* Fix div error in low pytorch version

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-21 13:27:04 +08:00
Ezra-Yu 789884bf08
[Refactor] Fix visualization tools. (#1045)
* update browse dataset

* update images

* update tools

* update grad_cam

* update docs

* update tools

* update docs
2022-09-20 18:09:05 +08:00
Ezra-Yu f1d2f50c21
[Feature] Add MobileOne Backbone For MMCls 1.x. (#1030)
* add mobileOne

* add train cfg

* update cfgs

* update URL

* update configs

* update inceptionv3 metafile

* add configs

* fix lint

* update checkpoint urls

* Update configs

* Update README

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-16 11:55:10 +08:00
Ma Zerun 9999da646c
[Imporve] Update tools to enable `pin_memory` and `persistent_workers` by default. (#1024) 2022-09-14 11:57:32 +08:00
Songyang Zhang 4367d05dfc
[Enhance] Use `fvcore` to calculate FLOPS. (#1000)
* [Feature] Use fvcore for flops count

* update requirements

* update
2022-08-31 11:04:42 +08:00
Ezra-Yu 0e3e8e91cc
[Refactor] Refactor Config Doc. (#987)
* refactor config doc

* refine

* Fix the config docs.

* update CN config

* Fix some grammar erros.

* Fix Chinese docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-08-29 11:10:05 +08:00
Ma Zerun 2b88df4484
[Refactor] Refactor the data flow. (#989)
* [Refactor] Refactor the data flow.

* Add comments about data preprocessor.

* Fix after mmengine folder structure refactoring.
2022-08-26 10:40:43 +08:00
Ma Zerun b7d0d521eb
[Refactor] Use `mmengine` instead of `mmcv` and refactor some transforms. (#986)
* [Refactor] Refactor the rest data transforms and fix some docstring.

* Use `mmengine` instead of `mmcv`

* Refactor KFold dataset tools

* Fix docstring according to comments
2022-08-24 15:59:02 +08:00
Ma Zerun b8b31e9343
[Improve] Multiple fixes. (#957)
* Rename some files in unit tests.

* Rename `pipelines` folder to `transforms`.

* Fix version requirement range.

* Remove `init_random_seed` and `set_random_seed`.

* Fix unit tests

* Rename `convert_models` to `model_converters`.
2022-08-09 15:58:08 +08:00
Ma Zerun 7f0502867b
[Improve] Add `--amp`, `--resume` and `--auto-scale-lr` options in `tools/train.py`. (#914) 2022-07-21 17:34:04 +08:00
Ezra-Yu 1c82863449 fix lint 2022-07-18 11:11:13 +08:00
Ezra-Yu 6ed3860ea6 refactor UTs 2022-07-18 11:11:13 +08:00
Ezra-Yu 0f30c392a3 from core to engine 2022-07-18 11:11:13 +08:00
mzr1996 24bcf069f8 [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
Ezra-Yu d8f556668e Refactor vis scheduler tool 2022-07-18 11:11:13 +08:00
mzr1996 5a3e736b4e [Feature] Implement offline evaluation. 2022-07-18 11:11:13 +08:00
Ezra-Yu a82de04b67 Enhance browse datset 2022-07-18 11:11:13 +08:00
mzr1996 ee6e585e41 Mazerun/base classifier 2022-07-18 11:11:13 +08:00
mzr1996 522ab1fd84 [Feature] Add Visualization hook 2022-07-18 11:11:04 +08:00
Ezra-Yu 4d4b22377d Add browse dataset 2022-07-18 10:53:56 +08:00
mzr1996 efaa93aae1 [Refactor] Refactor entry scripts `train.py` and `test.py`. 2022-07-18 10:53:56 +08:00
mzr1996 0537c4d70c [Feature] Add `ClsVisualizer`. 2022-07-18 10:53:27 +08:00
yingfhu de002e455f [Refactor] Use `MMLogger` from MMEngine in `get_logger` and `print_log`. 2022-07-18 10:53:27 +08:00
mzr1996 877ea3080a [Fix] Fix wrong `--local_rank`. 2022-05-17 21:10:25 +08:00
Hu Di b4eefe4794
[Enhance] Support training on IPU and add fine-tuning configs of ViT. (#723)
* implement training and evaluation on IPU

* fp16 SOTA

* Tput reaches 5600

* 123

* add poptorch dataloder

* change ipu_replicas to ipu-replicas

* add noqa to config long line(website)

* remove ipu dataloder test code

* del one blank line in test_builder

* refine the dataloder initialization

* fix a typo

* refine args for dataloder

* remove an annoted line

* process one more conflict

* adjust code structure in mmcv.ipu

* adjust ipu code structure in mmcv

* IPUDataloader to IPUDataLoader

* align with mmcv

* adjust according to mmcv

* mmcv code structre fixed

Co-authored-by: hudi <dihu@graphcore.ai>
2022-04-29 22:22:19 +08:00
takuoko df6edd7f5a
[Feature] Support VAN. (#739)
* add van

* fix config

* add metafile

* add test

* model convert script

* fix review

* fix lint

* fix the configs and improve docs

* rm debug lines

* add VAN into api

Co-authored-by: Yu Zhaohui <1105212286@qq.com>
2022-04-28 09:35:17 +08:00
Ma Zerun 02c8f82147
[Enhance] Add extra dataloader settings in configs. (#752)
* Use `train_dataloader`, `val_dataloader` and `test_dataloader` settings
in the `data` field to specify different arguments.

* Fix bug

* Fix bug
2022-04-01 15:57:12 +08:00
Ezra-Yu 922a31a68e
[Feature] ADD RepMLP Backbone and checkpoints. (#709)
* add repmlp

* refactor backbone code

* add checkpoint, readme, unit tests

* unchange demo

* improve docstring.

* update pic

* refactor patch embed

* refactor unit tests

* fix lint

* update tools

* update tools

* Update checkpoint path

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-30 20:40:25 +08:00
Ma Zerun a87ff7e153
[Enhance] Use `--a-b` instead of `--a_b` in arguments. (#754) 2022-03-30 19:16:28 +08:00
Hubert 995f2912ab
[Feature] Add diff seeds to diff ranks. (#744)
* [Feature] Add diff seeds to diff ranks.

* lint
2022-03-25 15:31:13 +08:00