Commit Graph

426 Commits (4fb44f8770159b70f324843c142b19017ce3e945)

Author SHA1 Message Date
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
Ma Zerun 940a06f645
[Refactor] Refactor to use new fileio API in MMEngine. (#1176)
* [Refactor] Refactor to use new fileio API in MMEngine.

* Add comment about why use `backend`
2022-11-21 10:56:35 +08:00
Hubert 629f6447ef
[Feature] Migrate CSRA head to 1.x. (#1177)
* [Feat] add csra to 1x

* minor fix

* add voc metrics

* refine

* add unittest

* minor fix

* add more comments

* Fix docs and metafile.

* Fix docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-21 10:39:16 +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
Ezra-Yu c3c1cb93aa
[Feature] Add Switch Recipe Hook. (#1101)
* add switch hook and UTs

* update doc

* update doc

* fix lint

* fix ci

* fix ci

* fix typo

* fix ci

* switchTrainAugHook to switchRecipeHook

* fix lint

* Refactor the `SwitchRecipeHook`.

* Fix windows CI

* Fix windows CI

* Fix windows CI.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-18 18:12:03 +08:00
XingyuXie e51ecfb129
[Feature] Add adan optimizer. (#1180)
* add adan optimizer

* uppdate adan optimizer

* uppdate adan optimizer

* uppdate adan optimizer

* uppdate adan optimizer

* uppdate adan optimizer

* uppdate adan optimizer

* uppdate adan optimizer

* uppdate adan optimizer

* uppdate adan optimizer

* uppdate init
2022-11-17 08:11:25 +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
Hubert 992d13e772
[Enhance] add deleting params info in swinv2. (#1142) 2022-11-14 17:07:21 +08:00
Hubert 1b98fc13d9
[Enhance] Add more mobilenetv3 pretrains. (#1154)
* add small-050/075 and move files

* add previous results

* Update checkpoint link

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-14 17:06:39 +08:00
Hakjin Lee cf5879988d
[Feature] Support Activation Checkpointing for ConvNeXt. (#1153)
* Support Activation Checkpointing for ConvNeXt

* Add docstring.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-14 15:04:28 +08:00
Ma Zerun 542143cb41
[Feature] Add TIMM and HuggingFace wrappers to build classifiers from them directly. (#1102)
* [Feature] Add TIMM and HuggingFace wrappers to build classifiers from them directly.

* Support `with_cp` and add docstring.

* Add unit tests.

* Update CI.

* Update docs.
2022-11-10 14:56:19 +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
zzc98 9eb6fc4368
[Feature] Add reduction for neck (#978)
* feat: add reduction for neck

* feat: add reduction for neck

* feat: add reduction for neck

* feat:add linear reduction neck

* feat: add reduction neck

* mod out of linearReduction as tuple

* fix typo

* fix unit tests

* fix unit tests

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-11-04 15:44:37 +08:00
takuoko 8cc1fdef52
[Enhancement] RepVGG for YOLOX-PAI for dev-1.x. (#1126) 2022-11-04 15:36:18 +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
Ezra-Yu 63b124e2d7
[Docs] Improve Swin Doc and Add Tabs enxtation. (#1145)
* improve_swin_doc

* fix requirments

* improve swin2 docs

* improve swin2 docs

* update

* update CN doc

* update CN doc

* update comments

* fix error

* update register_all_modules

* Update README.md

* Update stat.py

* Update readme

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-04 11:58:39 +08:00
zzc98 9506241f73
[Feature] Add arcface head. (#926)
* feat: add arcface head

* feat: add arcface head

* update arcface

* refactor archface head

* update archface head

* update archface head

* use torch.cat instead of torch.hstack to fix ci

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-11-02 17:45:33 +08:00
zzc98 693596bc2f
[Feature] Add Base Retriever and Image2Image Retriever for retrieval tasks. (#1055)
* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* feat: add image retriever

* update retriever

* fix lint

* add hook unit test

* Use `register_buffer` to save prototype vectors and add a progress bar
during preparing prototype.

* update UTs

* update UTs

* fix typo

* modify the hook

Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-02 17:43:56 +08:00
Hubert 50aaa711ea
[Docs] Add custom evaluation docs (#1130)
* [Docs] Add evaluation docs

* minor fix

* Fix docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-01 18:54:06 +08:00
mzr1996 b526f018db [Fix] Fix broken inference api because of the modification in data
preprocessor.
2022-10-18 17:50:35 +08:00
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
kitecats 06c919efc2
[Fix] Fix for `inference_model` cannot get classes information in checkpoint. (#1093)
* Fix for MMCLS1.x not being able to get classes information in checkpoint during inference

Let MMCLS1.x get classes information from checkpoint during inference instead of using imagenet classes initialization

* Update inference.py
2022-10-14 08:27:01 +08:00
Ma Zerun 31c67ffed4
Bump version to v1.0.0rc2. (#1084) 2022-10-12 16:52:31 +08:00
Ma Zerun dfe0874102
Update requirements. (#1083) 2022-10-12 10:57:37 +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 043574cbb2 Merge branch '1.x' into dev-1.x 2022-10-10 11:18:05 +08:00
mzr1996 2153a16dc5 [Docs] Fix some docstrings. 2022-10-10 11:17:48 +08:00
takuoko a1642e42da
[Enhancement] Get scores from inference api. (#1070) 2022-10-08 15:21:34 +08:00
mzr1996 ae37d7fd27 [Fix] Fix visualization hook. 2022-10-08 11:29:18 +08:00
Ma Zerun 7237a646e8
Bump version to v1.0.0rc1 (#1066) 2022-09-30 17:35:44 +08:00
whcao 8c5d86a388
[Fix] Fix `loss_weight` in `LabelSmoothLoss`. (#1058) 2022-09-28 08:18:12 +08:00
Ma Zerun 51b7c754e7 [Fix] Fix the output position of Swin-Transformer. (#947)
* [Fix] Fix the output position of Swin-Transformer.

* Rename `downsample` argument to `do_downsample`.
2022-09-22 17:11:52 +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
Yuan Liu f7191ca686
[Feature] Add ViT huge architecture. (#1049)
* [Feature]: Add ViT huge settings

* [Feature]: Add comments
2022-09-21 13:27:54 +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
mzr1996 b35ee5d139 [Fix] Fix wrong default CLASSES settings in `init_model`. 2022-09-20 17:48:03 +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 61e9d890a6
[CI] Update circle-ci and github workflow. (#1018)
* Add deploy workflow.

* [CI] Update circle-ci and github workflow.

* Fix windows CI

* Update unit tests to save memory
2022-09-13 11:46:15 +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
Ma Zerun 0b61b76325
[Docs] Add Migration documentation. (#997)
* [Docs] Add Migration documentation.

* Fix the train_cfg argument

* [WIP] Improve according to comments

* Imporve according to comments.
2022-08-31 13:31:48 +08:00
Ezra-Yu e34588a315
[Docs] Refactor API docs. (#958)
* rebase

* update process

* add TODOs in moved classes

* update APIs

* modify the preprocess_cfg to data_preprocessor

* Fix api docs.

Co-authored-by: mzr1996 <mzr1996@163.com>
2022-08-29 15:35:10 +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
Yixiao Fang 5665b8349a
[Refactor] Refactor import dependency. (#977)
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-08-22 15:17:33 +08:00
Ma Zerun 20f9ace5c5
[CI] Add Circle CI for mmcls 1.x. (#970)
* [CI] Add Circle CI for mmcls 1.x

* Fix circle-ci config

* Use `torch.cat` instead of `torch.hstack` to compat with PyTorch 1.6

* Compat the behavior of LongTensor in PyTorch 1.6

* Avoid random behavior in the unit test of Lighting

* Fix cuda ci

* Remove github workflow temporarily.
2022-08-22 15:02:08 +08:00