mirrors
/
mmpretrain
mirror of https://github.com/open-mmlab/mmpretrain.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
mmpretrain/model-index.yml

51 lines
1.6 KiB
YAML
Raw Normal View History

[Feature] Adding PWC metafile (#225) * add yml * fix VGG * minor fix * add model zoo * add bs and epoch * revised according to comments * revised according to comments * revised according to comments
2021-04-28 17:07:26 +08:00
Import:
- configs/mobilenet_v2/metafile.yml
[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
- configs/mobilenet_v3/metafile.yml
[Feature] Adding PWC metafile (#225) * add yml * fix VGG * minor fix * add model zoo * add bs and epoch * revised according to comments * revised according to comments * revised according to comments
2021-04-28 17:07:26 +08:00
- configs/resnet/metafile.yml
[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
- configs/res2net/metafile.yml
[Feature] Adding PWC metafile (#225) * add yml * fix VGG * minor fix * add model zoo * add bs and epoch * revised according to comments * revised according to comments * revised according to comments
2021-04-28 17:07:26 +08:00
- configs/resnext/metafile.yml
- configs/seresnet/metafile.yml
- configs/shufflenet_v1/metafile.yml
- configs/shufflenet_v2/metafile.yml
[Docs] Add algorithm readme and update meta yml (#418) * Add README.md for models without checkpoints. * Update model-index.yml * Update metafile.yml of seresnet
2021-08-24 17:46:46 +08:00
- configs/swin_transformer/metafile.yml
[Feature] Adding PWC metafile (#225) * add yml * fix VGG * minor fix * add model zoo * add bs and epoch * revised according to comments * revised according to comments * revised according to comments
2021-04-28 17:07:26 +08:00
- configs/vgg/metafile.yml
[Feature] Add RepVGG backbone and checkpoints. (#414) * Add RepVGG code. * Add se_module as plugin. * Add the repvggA0 primitive config * Change repvggA0.py to fit mmcls * Add RepVGG configs * Add repvgg_to_mmcls * Add tools/deployment/convert_repvggblock_param_to_deploy.py * Change configs/repvgg/README.md * Streamlining the number of configuration files. * Fix lints * Delete plugins * Delete code about plugin. * Modify the code for using se module. * Modify config to fit repvgg with se. * Change se_cfg to allow loading of pre-training parameters. * Reduce the complexity of the configuration file. * Finsh unitest for repvgg. * Fix bug about se in repvgg_to_mmcls. * Rename convert_repvggblock_param_to_deploy.py to reparameterize_repvgg.py, and delete setting about device. * test commit * test commit * test commit command * Modify repvgg.py to make the code more readable. * Add value=0 in F.pad() * Add se_cfg to arch_settings. * Fix bug. * modeify some attr name and Update unit tests * rename stage_0 to stem and branch_identity to branch_norm * update unit tests * add m.eval in unit tests * [Enhance] Enhence SE layer to support custom squeeze channels. (#417) * add enhenced SE * Update * rm basechannel * fix docstring * Update se_layer.py fix docstring * [Docs] Add algorithm readme and update meta yml (#418) * Add README.md for models without checkpoints. * Update model-index.yml * Update metafile.yml of seresnet * [Enhance] Add `hparams` argument in `AutoAugment` and `RandAugment` and some other improvement. (#398) * Add hparams argument in `AutoAugment` and `RandAugment`. And `pad_val` supports sequence instead of tuple only. * Add unit tests for `AutoAugment` and `hparams` in `RandAugment`. * Use smaller test image to speed up uni tests. * Use hparams to simplify RandAugment config in swin-transformer. * Rename augment config name from `pipeline` to `pipelines`. * Add some commnet ad docstring. * [Feature] Support classwise weight in losses (#388) * Add classwise weight in losses:CE,BCE,softBCE * Update unit test * rm some extra code * rm some extra code * fix broadcast * fix broadcast * update unit tests * use new_tensor * fix lint * [Enhance] Better result visualization (#419) * Imporve result visualization to support wait time and change the backend to matplotlib. * Add unit test for visualization * Add adaptive dpi function * Rename `imshow_cls_result` to `imshow_infos`. * Support str in `imshow_infos` * Improve docstring. * Bump version to v0.15.0 (#426) * [CI] Add PyTorch 1.9 and Python 3.9 build workflow, and remove some CI. (#422) * Add PyTorch 1.9 build workflow, and remove some CI. * Add Python 3.9 CI * Show Python 3.9 support. * [Enhance] Rename the option `--options` in some tools to `--cfg-options`. (#425) * [Docs] Fix sphinx version (#429) * [Docs] Add `CITATION.cff` (#428) * Add CITATION.cff * Fix typo in setup.py * Change author in setup.py * modeify some attr name and Update unit tests * rename stage_0 to stem and branch_identity to branch_norm * update unit tests * add m.eval in unit tests * Update unit tests * refactor * refactor * Alignment inference accuracy * Update configs, readme and metafile * Update readme * return tuple and fix metafile * fix unit test * rm regnet and classifiers changes * update auto_aug * update metafile & readme * use delattr * rename cfgs * Update checkpoint url * Update readme * Rename config files. * Update readme and metafile * add comment * Update mmcls/models/backbones/repvgg.py Co-authored-by: Ma Zerun <mzr1996@163.com> * Update docstring * Improve docstring. * Update unittest_testblock Co-authored-by: Ezra-Yu <1105212286@qq.com> Co-authored-by: Ma Zerun <mzr1996@163.com>
2021-09-29 11:06:23 +08:00
- configs/repvgg/metafile.yml
[Docs] Update metafile and Readme (#435) * Update metafile format. * Update accuracy of checkpoints. * Add metafile and readme for tnt. * Add converted ckpts in swin-transformer. * Fix tnt ckpt link * Update swin_transformer metafile
2021-09-18 16:32:46 +08:00
- configs/tnt/metafile.yml
[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
- configs/vision_transformer/metafile.yml
[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
- configs/t2t_vit/metafile.yml
[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
- configs/tinyvit/metafile.yml
[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
- configs/mlp_mixer/metafile.yml
[Feature] Implement the conformer backbone. (#494) * implement the conformer * format code style * format code style * reuse the TransformerEncoderLayer in the vision_transformer.py * Modify variable name * delete unused params * Remove warning info in Conformer head since it already exists in Conformer. * Rename some variables * Add unit tests * Use `getattr` instead of `get_submodule`. * Remove some useless layers * Refactor conformer and add configs * Update configs and add metafile. * Fix unit tests * Update README Co-authored-by: mzr1996 <mzr1996@163.com>
2021-12-07 14:00:17 +08:00
- configs/conformer/metafile.yml
[Reproduction] Reproduce RegNetX training accuracy. (#587) * add preciseBN hook * add precise-bn config * add proority of preciseBN * Add precise BN hook * Update PreciseBN hook * add config example & pre-commit * Update PBN * Add unit test * Update unit tests * mv preciseBN to mmcls.runner * rm extra sleep() * merge latest * update code * fix lint * update unit tests * add regnetx configs * update prcise BN in configs * fix lint * add checkpoints and logs * fix conflict * fix lint * update model_zoo and model-index * remove some changes * update readme.md * remove precise_bn.py
2021-12-28 17:34:27 +08:00
- configs/regnet/metafile.yml
[Feature] Add DeiT backbone and checkpoints. (#576) * Support DeiT backbone. * Use hook to automatically resize pos embed * Update ViT training setting * Add deit configs and update docs * Fix vit arch assertion * Remove useless init function * Add unit tests. * Fix resize_pos_embed for DeiT * Improve according to comments.
2021-12-15 22:44:57 +08:00
- configs/deit/metafile.yml
[Feature] Add Twins backbone and convert checkpoints. (#642) * add twins backbone * add position_encoding * refactor twins * Supplemental unit tests * update docstring and readme * update docstring and readme * update docstring and readme * update docstring * update docstring * update docstring * update docstring * remove note * update doc and docstring * update docstring * update docstring * use abstract pdf link and rename yamlfile * Update model link Co-authored-by: mzr1996 <mzr1996@163.com>
2022-01-27 10:25:05 +08:00
- configs/twins/metafile.yml
[Feature] Support EfficientNet (#649) * add config for resnest test * fix config * add label smoothing * add memcached * minor fix * fix bug * fix config * add config * minor fix * fix configs * use EResize * change interpolation * add more configs * add docsting * add unittest * remove unnecessary changes * minor fix * add more docstring * fix linting * add efficient backbone * add config * add Edge Residual * fix bug * remove unnecessary files * refactor * add resize in crop to ensure crop size is output size * fix bug and add comments * test * fix * add more configs * add more configs * add more configs * fix bug * add model zoo * fix * reorganize code * add edge tpu * add edge tpu converter * rename * update readme * reorganize code and config * Rename configs of EfficientNet, and add metafile & model_zoo * Remove `backend='pillow'` * Add comments about EfficientNet-EdgeTPU * Rename the convert tool of EfficientNet. * Refactor EfficientNet and update docstring. * Update EfficientNet-EdgeTPU config * Fix unit tests Co-authored-by: lixinran <lixr423@outlook.com> Co-authored-by: lixinran <lixinran@sensetime.com> Co-authored-by: mzr1996 <mzr1996@163.com>
2022-01-25 12:14:17 +08:00
- configs/efficientnet/metafile.yml
[Feature] Support ConvNeXt (#670) * Support ConvNeXt * Add configs of ConvNeXt * Update dev scripts * Update docs. * Use new style README * Add unit tests. * Update README * Imporve according to comments * Modify refers to timm. * Imporve according to comments
2022-01-28 10:36:45 +08:00
- configs/convnext/metafile.yml
[Feature] Support HRNet and add pre-trained models (#660) * Support HRNet * Add HRNet configs * Fix a bug in backward * Add configs and update docs. * Not use bias in conv before batch norm * Defaults to use `norm_eval=False` * Add unit tests and support out_channels in HRFuseScales * Update checkpoint path * Update docstring. * Remove incorrect files * Improve according to comments
2022-01-28 10:54:14 +08:00
- configs/hrnet/metafile.yml
[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
- configs/repmlp/metafile.yml
[Feature] Support Wide-ResNet. (#715) * Add configs of Wide ResNet * updated config * add docs and metafile * update configs * remove extra import * Update metafile and readme Co-authored-by: Ezra-Yu <1105212286@qq.com> Co-authored-by: mzr1996 <mzr1996@163.com>
2022-03-04 14:06:34 +08:00
- configs/wrn/metafile.yml
[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
- configs/van/metafile.yml
[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
- configs/cspnet/metafile.yml
[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
- configs/convmixer/metafile.yml
[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
- configs/densenet/metafile.yml
[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
- configs/poolformer/metafile.yml
[Refactor] Refactor dev scripts
2022-07-12 16:10:59 +08:00
- configs/inception_v3/metafile.yml
[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
- configs/mvit/metafile.yml
[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
- configs/edgenext/metafile.yml
[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
- configs/mobileone/metafile.yml
[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
- configs/efficientformer/metafile.yml
[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
- configs/swin_transformer_v2/metafile.yml
[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
- configs/deit3/metafile.yml
[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
- configs/hornet/metafile.yml
[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
- configs/mobilevit/metafile.yml
[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
- configs/davit/metafile.yml
[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
- configs/replknet/metafile.yml
[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
- configs/csra/metafile.yml
[Refactor] Refactor BEiT backbone and support v1/v2 inference. (#1144) * refactor beit backbone * use LinearClsHead * fix mean and std value * fix lint * support inference if beit-v2 * update encoder layer and init * update * add ut * add prepare_relative_position_bias_table function * add cls_token * fix lint * add pos_embed check * update metafile and readme * update weights link * update link of weights * update metafile * update * update docstrings * update according to review * rename readme * update docstring * fix lint
2022-11-29 12:56:33 +08:00
- configs/beit/metafile.yml
- configs/beitv2/metafile.yml
[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
- configs/eva/metafile.yml
[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
- configs/revvit/metafile.yml
[Feature] Add clip backbone. (#1258) * clip backbone added * passed precommit * update readme * update according to PR review * add missing file * add unittest * refine metafile * refine metafile and readme for readdocs * refine metafile * refine metafile * Update metafile Co-authored-by: mzr1996 <mzr1996@163.com>
2022-12-22 16:33:57 +08:00
- configs/clip/metafile.yml
[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
- configs/mixmim/metafile.yml
[Feature] [CodeCamp #68] Add EfficientnetV2 Backbone. (#1253) * add efficientnet_v2.py * add efficientnet_v2 in __init__.py * add efficientnet_v2_s base config file * add efficientnet_v2 config file * add efficientnet_v2 config file * update tuple output * update config file * update model file * update model file * update model file * update config file * update model file * update config file * update model file * update model file * update model file * update model file * update model file * update config file * update config file * update model file * update model file * update model file * update model file * update model config file * Update efficientnet_v2.py * add config file and modify arch * add config file and modify arch * add the file about convert_pth from timm to mmcls * update efficientnetv2 model file with mmcls style * add the file about convert_pth from timm to mmcls * add the file about convert_pth from timm to mmcls * update convert file * update model file * update convert file * update model file * update model file * update model file * add metefile and README * Update tools/model_converters/efficientnetv2-timm_to_mmcls.py Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com> * update model file and convert file * update model file and convert file * update model file and convert file * update model file and convert file * update model file * update model file * update model file * update config file and README file * update metafile * Update efficientnetv2_to_mmcls.py * update model-index.yml * update metafile.yml * update b0 and s train pipeline * update b0 and s train pipeline * update b0 and s train pipeline * add test_efficientnet_v2 * update test_efficientnet_v2 * update model file docs * update test_efficientnet_v2 * update test_efficientnet_v2 * add efficientnet_v2.py * add efficientnet_v2 in __init__.py * add efficientnet_v2_s base config file * add efficientnet_v2 config file * add efficientnet_v2 config file * update tuple output * update config file * update model file * update model file * update model file * update model file * update config file * update config file * update model file * update model file * update model file * update model file * update model file * update config file * update config file * update model file * update model file * update model file * update model file * update model config file * Update efficientnet_v2.py * add config file and modify arch * add config file and modify arch * add the file about convert_pth from timm to mmcls * update efficientnetv2 model file with mmcls style * add the file about convert_pth from timm to mmcls * add the file about convert_pth from timm to mmcls * update convert file * update model file * update convert file * update model file * update model file * update model file * add metefile and README * Update tools/model_converters/efficientnetv2-timm_to_mmcls.py Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com> * update model file and convert file * update model file and convert file * update model file and convert file * update model file and convert file * update model file * update model file * update model file * update config file and README file * update metafile * Update efficientnetv2_to_mmcls.py * update model-index.yml * update metafile.yml * update b0 and s train pipeline * update b0 and s train pipeline * update b0 and s train pipeline * add test_efficientnet_v2 * update test_efficientnet_v2 * update model file docs * update test_efficientnet_v2 * update test_efficientnet_v2 * pass pre-commit hook * refactor efficientnetv2 * refactor efficientnetv2 * update readme, metafile and weight links * update model-index.yml * fix lint * fix typo * Update efficientnetv2-b1_8xb32_in1k.py * Update efficientnetv2-b2_8xb32_in1k.py * Update efficientnetv2-b3_8xb32_in1k.py * update two moduals and model file * update modual file * update accuracys * update accuracys * update metafile * fix build docs * update links * update README.md Co-authored-by: qingtian <459291290@qq.com> Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2022-12-30 15:18:39 +08:00
- configs/efficientnet_v2/metafile.yml