Commit Graph

147 Commits (bcca61906649dcfb91a11459175cfa0578cd27f3)

Author SHA1 Message Date
Hubert bcca619066
[Feature] Support MobileViT backbone. (#1068)
* init

* fix

* add config

* add meta

* add unittest

* fix for comments

* Imporvee docstring and support custom arch.

* Update README

* Update windows CI

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-10-18 17:05:59 +08:00
Ma Zerun 29f066f7fb
[Improve] Speed up data preprocessor. (#1064)
* [Improve] Speed up data preprocessor.

* Add ClsDataSample serialization override functions.

* Add unit tests

* Modify configs to fit new mixup args.

* Fix `num_classes` of the ImageNet-21k config.

* Update docs.
2022-10-17 17:08:18 +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
mzr1996 d90dfc3d99 [Docs] Use relative link to config instead of abs link in README. 2022-09-22 09:59:06 +08:00
Ma Zerun 034683cfcd
[Feature] Support MViT for MMCLS 1.x (#1023)
* [Feature] Support MViT for MMCLS 1.x

* Add unit tests

* Update documentation
2022-09-22 09:23:49 +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
Ma Zerun e9e2d48cb2
[Improve] Update benchmark scripts (#1028)
* Update train benchmark scripts

* Add `--cfg-options` for dev scripts and enhance `--range`.

* Fix bug of regex expression.

* Fix minor bugs

* Update ShuffleNet configs

* Update rsb-a1 configs and label smooth loss mode.

* Update inference dev scripts

* From `mmengine` instead of `mmcv` import fileio.

* Fix lint

* Update pre-commit hook

* Use `use_sigmoid` option instead of "bce" mode in label smooth loss.
2022-09-20 15:50:21 +08:00
Hubert e4e8047563
[Feature] Support Swin Transformer V2 for MMCLS 1.x. (#1029)
* [Feature] support swin v2 in mmcls 1.x

* fix configs

* fix clamp max

* Fix meshgrid

* Imporve swin v2

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-20 15:45:27 +08:00
Ezra-Yu aaf127c5e1
[Feature] Add efficientformer Backbone for MMCls 1.x. (#1031)
* rebase

* update filename

* update URL

* update UT

* fix lint

* update head

* add efficientformer

* update filename

* update UT

* fix lint

* update configs

* rebase

* fix unit tests

* Fix comments and docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-09-20 14:56:45 +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 96a1a34415
[Docs] Auto generate model summary table. (#1010)
* Fix wrong metafile

* Auto generate model summary table.

* Fix all TODO link
2022-09-13 15:06:17 +08:00
Ma Zerun 85b1eae7f1
Bump to v1.0.0rc0 (#1007)
* Update docs.

* Update requirements.

* Update config readme and docstring.

* Update CONTRIBUTING.md

* Update README

* Update requirements/mminstall.txt

Co-authored-by: Yifei Yang <2744335995@qq.com>

* Update MMEngine docs link and add to readthedocs requirement.

Co-authored-by: Yifei Yang <2744335995@qq.com>
2022-08-31 23:57:51 +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
Ezra-Yu 624d22b441
[Refactor] Refactor Precise BN Hook. (#913) 2022-08-03 15:24:19 +08:00
mzr1996 0c7a04b1c7 Fix lint 2022-07-18 14:14:31 +08:00
mzr1996 735a3ee11f Update auto_scale_lr fields 2022-07-18 11:11:13 +08:00
yingfhu a667b488ae minor fix 2022-07-18 11:11:13 +08:00
yingfhu ce81a07059 [Refactor] add auto_scale_lr 2022-07-18 11:11:13 +08:00
mzr1996 24bcf069f8 [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
mzr1996 c0f3ba68a8 Add MAE version ViT-base training results and fix some errors in configs. 2022-07-18 11:11:13 +08:00
Ezra-Yu d8f556668e Refactor vis scheduler tool 2022-07-18 11:11:13 +08:00
Ezra-Yu 7bca2516f0 Add CUB dataset 2022-07-18 11:11:13 +08:00
Ezra-Yu a9057e88c4 Add multi label metrics 2022-07-18 11:11:13 +08:00
mzr1996 f3299b4ca2 [Refactor] Refactor batch augmentations 2022-07-18 11:11:13 +08:00
yingfhu 125b74d4ca [Refactor] Refactor MNIST and FashionMNIST dataset. 2022-07-18 11:11:13 +08:00
mzr1996 522ab1fd84 [Feature] Add Visualization hook 2022-07-18 11:11:04 +08:00
yingfhu c78b5597d8 [Refactor] refactor randomCrop, randomResizeCrop and CenterCrop。 2022-07-18 10:57:17 +08:00
mzr1996 6563f5f448 Update autoaug and randaug configs 2022-07-18 10:53:56 +08:00
mzr1996 548db6f4ac [Refactor] Update optimizer related registries and configs. 2022-07-18 10:53:56 +08:00
mzr1996 995b1d0d58 [Refactor] Add `ResizeEdge` and refactor all dataset configs. 2022-07-18 10:53:56 +08:00
Ezra-Yu 4d4b22377d Add browse dataset 2022-07-18 10:53:56 +08:00
yingfhu 65f3b2221d Modify config relates to logger and checkpoint 2022-07-18 10:53:28 +08:00
Ezra-Yu 58d9f649ed Refactor scheduler configuration 2022-07-18 10:53:28 +08:00
Ezra-Yu 2f2aa3037c Refactor default hooks configs 2022-07-18 10:53:28 +08:00
Ezra-Yu 93a27c8324 [Feature] Add `PackClsInputs` and use `LoadImageFromFile`, `Resize` & `RandomFlip` in MMCV. 2022-07-18 10:53:28 +08:00
mzr1996 58b21ee56f [Model] Add IPU ViT model 2022-06-03 18:15:28 +08:00
Ma Zerun 824fbcbaae
[Refactor] Use mdformat instead of markdownlint to format markdown. (#844)
* [Refactor] Use mdformat instead of markdownlint to format markdown.

* Update unavailiable api links in tutorials

* Update CONTRIBUTING.md

* Use mdformat==0.7.9 to support Python 3.6
2022-06-02 15:22:01 +08:00
Ma Zerun 7c5ddb1e5b
Bump version to v0.23.0 (#809) 2022-05-01 21:58:33 +08:00
mzr1996 1d6fbe0efe [Fix] Fix lint and mmcv version requirement for IPU. 2022-04-29 22:33:29 +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
Hubert 59292b315a
[Feature] Support DenseNet. (#750)
* init add densenet implementation

* Add config and converted models

* update meta

* add test for memory efficient

* Add docs

* add doc for jit

* Update checkpoint path

* Update readthedocs

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-04-28 13:06:03 +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
mzr1996 29b882d4eb Merge branch 'dev' 2022-04-15 20:10:45 +08:00
Ma Zerun eda75d12ec
Bump version to v0.22.1 (#785) 2022-04-15 20:08:39 +08:00
Weihao Yu 2eb27d0d4e
[Feature] Add PoolFormer backbone and checkpoints. (#746)
* add PoolFormer

* fix some typos in PoolFormer

* fix lint error

* modify out_indices and gap

* fix typo

* fix lint

* fix typo

* fix typo in poolforemr README

* fix lint

* Update some paths

* Refactor freeze_stages method

* Add unit tests

* Fix lint

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-04-15 11:35:15 +08:00
mzr1996 f0ee5dcb2a Update checkpoint path of ResNet-50 on CUB dataset 2022-03-31 10:23:33 +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
Ezra-Yu 292127098f
[Feature] Add CSPNet and backbone and checkpoints (#735)
* add cspnet backbone

* try

* add backbones

* add cspnet

* add acc

* add unit tests

* chnage mode

* Update checkpoint path

* Fix typo

* Imporve CSPNet

* Update checkpoints

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-30 15:29:35 +08:00
Hubert 04cb42a768
[Feature] Support ConvMixer. (#716)
* basic support for ConvMixer

* simplify

* add data pipeine config for timm

* Add model readme and metafile

* add unittest for convmixer

* add copyright

* modify

* add tests

* update model

* add conv2dAdaptivePadding replacement

* update model index

* fix comments

* Update checkpoint path

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-23 15:23:57 +08:00
takuoko aa522f4309
[Feature] Support CUB dataset. (#703)
* support cub dataset

* support cub dataset

* fix train lint error

* add docs

* fix class label

Co-authored-by: Ezra-Yu <1105212286@qq.com>

* del debug code

* skip docformatter problem

* add unit tests

* add CUB baseline configs and chpts

* fix some typos

* fix name style

* update flops

Co-authored-by: Ezra-Yu <1105212286@qq.com>
2022-03-16 16:22:28 +08:00