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

86 lines
2.7 KiB
YAML
Raw Permalink 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
update riformer mmpretrain
2023-04-04 13:17:16 +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
[Feature] Support ConvNeXt-V2 backbone. (#1294) * [Feature] Support ConvNeXt-V2. * Use registry of mmcls instead of mmengine. * Add README. * Add unit tests and docs.
2023-01-06 16:13:41 +08:00
- configs/convnext_v2/metafile.yml
[Feature] Support LeViT backbone. (#1238) * 网络搭建完成、能正常推理 * 网络搭建完成、能正常推理 * 网络搭建完成、能正常推理 * 添加了模型转换未验证，配置文件 但有无法运行 * 模型转换、结构验证完成，可以推理出正确答案 * 推理精度与原论文一致 已完成转化 * 三个方法改为class 暂存 * 完成推理精度对齐 误差0.04 * 暂时使用的levit2mmcls * 训练跑通，训练相关参数未对齐 * '训练相关参数对齐'参数' * '修复训练时验证导致模型结构改变无法复原问题' * '修复训练时验证导致模型结构改变无法复原问题' * '添加mixup和labelsmooth' * '配置文件补齐' * 添加模型转换 * 添加meta文件 * 添加meta文件 * 删除demo.py测试文件 * 添加模型README文件 * docs文件回滚 * model-index删除末行空格 * 更新模型metafile * 更新metafile * 更新metafile * 更新README和metafile * 更新模型README * 更新模型metafile * Delete the model class and get_LeViT_model methods in the mmcls.models.backone.levit file * Change the class name to Google Code Style * use arch to provide default architectures * use nn.Conv2d * mmcv.cnn.fuse_conv_bn * modify some details * remove down_ops from the architectures. * remove init_weight function * Modify ambiguous variable names * Change the drop_path in config to drop_path_rate * Add unit test * remove train function * add unit test * modify nn.norm1d to build_norm_layer * update metafile and readme * Update configs and LeViT implementations. * Update README. * Add docstring and update unit tests. * Revert irrelative modification. * Fix unit tests * minor fix Co-authored-by: mzr1996 <mzr1996@163.com>
2023-01-17 17:43:42 +08:00
- configs/levit/metafile.yml
[Feature] Support VIG Backbone. (#1304) * 添加vig源文件 * 某些模块修改到mmcls风格 * 修改到mmcls风格 * 修改 * 添加VIG模型及源文件 * update model file * update model file and config * change class name and some variable name * change class name and some variable name * update * update * change nn.BatchNorm to mmcv.cnn.build_norm_layer * update * change nn.Seq to mmcls * change backbone to stage_blocks * add vig_head * update * update config file * update * add readme and metafile * update model-index.yml * update model file * rename config file and add docstring * variable rename * update readme and metafile * update readme * update * Update VIG backbone implementation and docs. * Fix configs. Co-authored-by: mzr1996 <mzr1996@163.com>
2023-01-17 16:55:56 +08:00
- configs/vig/metafile.yml
[Feature] Support RetrieverRecall metric & Add ArcFace config (#1316) * rebase * add ap metric * fix mlti-gpu bug in retrevel * rebase * rebase * add training cfgs and update readme.md * fix bugs(cannot load vecs in dist and diff test-val recall\) * update configs and readme * fix ut * fix doc * rebase * fix rebase conflicts * fix rebase error * fix UT error * fix docs * fix typo --------- Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-02-14 12:46:21 +08:00
- configs/arcface/metafile.yml
[Feature] Support XCiT Backbone. (#1305) * update model file * Update XCiT implementation and configs. * Update metafiles * Update metafile * Fix floor divide * Imporve memory usage --------- Co-authored-by: qingtian <459291290@qq.com> Co-authored-by: mzr1996 <mzr1996@163.com>
2023-02-15 10:32:35 +08:00
- configs/xcit/metafile.yml
[Refactor] Refactor configs and metafile (#1369) * update base datasets * update base * update barlowtwins * update with new convention * update * update * update * add schedule * add densecl * add eva * add mae * add maskfeat * add milan and mixmim * add moco * add swav simclr * add simmim and simsiam * refine * update * add to model index * update config inheritance * fix error in metafile * Update pre-commit and metafile check script * update metafile * fix name error * Fix classification model name and config name --------- Co-authored-by: mzr1996 <mzr1996@163.com>
2023-02-23 11:17:16 +08:00
- configs/byol/metafile.yml
- configs/densecl/metafile.yml
- configs/mocov2/metafile.yml
- configs/mocov3/metafile.yml
- configs/simclr/metafile.yml
- configs/simsiam/metafile.yml
- configs/swav/metafile.yml
- configs/mae/metafile.yml
- configs/simmim/metafile.yml
- configs/barlowtwins/metafile.yml
- configs/cae/metafile.yml
- configs/maskfeat/metafile.yml
- configs/milan/metafile.yml
[Feature] Support multiple multi-modal algorithms and inferencers. (#1561) * [Feat] Migrate blip caption to mmpretrain. (#50) * Migrate blip caption to mmpretrain * minor fix * support train * [Feature] Support OFA caption task. (#51) * [Feature] Support OFA caption task. * Remove duplicated files. * [Feature] Support OFA vqa task. (#58) * [Feature] Support OFA vqa task. * Fix lint. * [Feat] Add BLIP retrieval to mmpretrain. (#55) * init * minor fix for train * fix according to comments * refactor * Update Blip retrieval. (#62) * [Feature] Support OFA visual grounding task. (#59) * [Feature] Support OFA visual grounding task. * minor add TODO --------- Co-authored-by: yingfhu <yingfhu@gmail.com> * [Feat] Add flamingos coco caption and vqa. (#60) * first init * init flamingo coco * add vqa * minor fix * remove unnecessary modules * Update config * Use `ApplyToList`. --------- Co-authored-by: mzr1996 <mzr1996@163.com> * [Feature]: BLIP2 coco retrieval (#53) * [Feature]: Add blip2 retriever * [Feature]: Add blip2 all modules * [Feature]: Refine model * [Feature]: x1 * [Feature]: Runnable coco ret * [Feature]: Runnable version * [Feature]: Fix lint * [Fix]: Fix lint * [Feature]: Use 364 img size * [Feature]: Refactor blip2 * [Fix]: Fix lint * refactor files * minor fix * minor fix --------- Co-authored-by: yingfhu <yingfhu@gmail.com> * Remove * fix blip caption inputs (#68) * [Feat] Add BLIP NLVR support. (#67) * first init * init flamingo coco * add vqa * add nlvr * refactor nlvr * minor fix * minor fix * Update dataset --------- Co-authored-by: mzr1996 <mzr1996@163.com> * [Feature]: BLIP2 Caption (#70) * [Feature]: Add language model * [Feature]: blip2 caption forward * [Feature]: Reproduce the results * [Feature]: Refactor caption * refine config --------- Co-authored-by: yingfhu <yingfhu@gmail.com> * [Feat] Migrate BLIP VQA to mmpretrain (#69) * reformat * change * change * change * change * change * change * change * change * change * change * change * change * change * change * change * change * change * change * change * refactor code --------- Co-authored-by: yingfhu <yingfhu@gmail.com> * Update RefCOCO dataset * [Fix] fix lint * [Feature] Implement inference APIs for multi-modal tasks. (#65) * [Feature] Implement inference APIs for multi-modal tasks. * [Project] Add gradio demo. * [Improve] Update requirements * Update flamingo * Update blip * Add NLVR inferencer * Update flamingo * Update hugging face model register * Update ofa vqa * Update BLIP-vqa (#71) * Update blip-vqa docstring (#72) * Refine flamingo docstring (#73) * [Feature]: BLIP2 VQA (#61) * [Feature]: VQA forward * [Feature]: Reproduce accuracy * [Fix]: Fix lint * [Fix]: Add blank line * minor fix --------- Co-authored-by: yingfhu <yingfhu@gmail.com> * [Feature]: BLIP2 docstring (#74) * [Feature]: Add caption docstring * [Feature]: Add docstring to blip2 vqa * [Feature]: Add docstring to retrieval * Update BLIP-2 metafile and README (#75) * [Feature]: Add readme and docstring * Update blip2 results --------- Co-authored-by: mzr1996 <mzr1996@163.com> * [Feature] BLIP Visual Grounding on MMPretrain Branch (#66) * blip grounding merge with mmpretrain * remove commit * blip grounding test and inference api * refcoco dataset * refcoco dataset refine config * rebasing * gitignore * rebasing * minor edit * minor edit * Update blip-vqa docstring (#72) * rebasing * Revert "minor edit" This reverts commit 639cec757c215e654625ed0979319e60f0be9044. * blip grounding final * precommit * refine config * refine config * Update blip visual grounding --------- Co-authored-by: Yiqin Wang 王逸钦 <wyq1217@outlook.com> Co-authored-by: mzr1996 <mzr1996@163.com> * Update visual grounding metric * Update OFA docstring, README and metafiles. (#76) * [Docs] Update installation docs and gradio demo docs. (#77) * Update OFA name * Update Visual Grounding Visualizer * Integrate accelerate support * Fix imports. * Fix timm backbone * Update imports * Update README * Update circle ci * Update flamingo config * Add gradio demo README * [Feature]: Add scienceqa (#1571) * [Feature]: Add scienceqa * [Feature]: Change param name * Update docs * Update video --------- Co-authored-by: Hubert <42952108+yingfhu@users.noreply.github.com> Co-authored-by: yingfhu <yingfhu@gmail.com> Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com> Co-authored-by: Yiqin Wang 王逸钦 <wyq1217@outlook.com> Co-authored-by: Rongjie Li <limo97@163.com>
2023-05-19 16:50:04 +08:00
- configs/ofa/metafile.yml
Bump version to v1.0.0rc7 (#1465) * update * update info * update changelog * update * update description * change to v1.0.0rc7
2023-04-07 17:34:21 +08:00
- configs/riformer/metafile.yml
[Feature] Add ViT of SAM (#1476) * add vit of sam * update * update * add ut * update ut * remove num_classes * support dynamic input * add ut * add comments * update ut
2023-04-13 17:03:28 +08:00
- configs/sam/metafile.yml
[Feature] Support GLIP (#1308) * rebase * add glip * update glip * add links * rename * fix doc --------- Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-04-17 19:19:23 +08:00
- configs/glip/metafile.yml
[Feature] add eva02 backbone (#1450) * [CI] Add test mim CI. (#879) * [CI] Add test mim CI. (#879) * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * update * update ci * rebase * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * update * update readme and configs * update readme and configs * refactore eva02 * [CI] Add test mim CI. (#879) * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * update * update ci * rebase * feat: add eva02 backbone * feat: add eva02 backbone * feat: add eva02 backbone * update * update readme and configs * refactore eva02 * update readme and metafile * update readme and metafile * update readme and metafile * update * rename eva02 * rename eva02 * fix uts * rename configs --------- Co-authored-by: Ma Zerun <mzr1996@163.com> Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
2023-05-06 19:28:31 +08:00
- configs/eva02/metafile.yml
[Feature] Support dinov2 backbone (#1522) * support dinov2 backbone * update metafile and readme * compatible to use_layer_scale * update SwiGLUFFN * add deprecation warning * update
2023-05-05 16:59:37 +08:00
- configs/dinov2/metafile.yml
[Feature] Support multiple multi-modal algorithms and inferencers. (#1561) * [Feat] Migrate blip caption to mmpretrain. (#50) * Migrate blip caption to mmpretrain * minor fix * support train * [Feature] Support OFA caption task. (#51) * [Feature] Support OFA caption task. * Remove duplicated files. * [Feature] Support OFA vqa task. (#58) * [Feature] Support OFA vqa task. * Fix lint. * [Feat] Add BLIP retrieval to mmpretrain. (#55) * init * minor fix for train * fix according to comments * refactor * Update Blip retrieval. (#62) * [Feature] Support OFA visual grounding task. (#59) * [Feature] Support OFA visual grounding task. * minor add TODO --------- Co-authored-by: yingfhu <yingfhu@gmail.com> * [Feat] Add flamingos coco caption and vqa. (#60) * first init * init flamingo coco * add vqa * minor fix * remove unnecessary modules * Update config * Use `ApplyToList`. --------- Co-authored-by: mzr1996 <mzr1996@163.com> * [Feature]: BLIP2 coco retrieval (#53) * [Feature]: Add blip2 retriever * [Feature]: Add blip2 all modules * [Feature]: Refine model * [Feature]: x1 * [Feature]: Runnable coco ret * [Feature]: Runnable version * [Feature]: Fix lint * [Fix]: Fix lint * [Feature]: Use 364 img size * [Feature]: Refactor blip2 * [Fix]: Fix lint * refactor files * minor fix * minor fix --------- Co-authored-by: yingfhu <yingfhu@gmail.com> * Remove * fix blip caption inputs (#68) * [Feat] Add BLIP NLVR support. (#67) * first init * init flamingo coco * add vqa * add nlvr * refactor nlvr * minor fix * minor fix * Update dataset --------- Co-authored-by: mzr1996 <mzr1996@163.com> * [Feature]: BLIP2 Caption (#70) * [Feature]: Add language model * [Feature]: blip2 caption forward * [Feature]: Reproduce the results * [Feature]: Refactor caption * refine config --------- Co-authored-by: yingfhu <yingfhu@gmail.com> * [Feat] Migrate BLIP VQA to mmpretrain (#69) * reformat * change * change * change * change * change * change * change * change * change * change * change * change * change * change * change * change * change * change * change * refactor code --------- Co-authored-by: yingfhu <yingfhu@gmail.com> * Update RefCOCO dataset * [Fix] fix lint * [Feature] Implement inference APIs for multi-modal tasks. (#65) * [Feature] Implement inference APIs for multi-modal tasks. * [Project] Add gradio demo. * [Improve] Update requirements * Update flamingo * Update blip * Add NLVR inferencer * Update flamingo * Update hugging face model register * Update ofa vqa * Update BLIP-vqa (#71) * Update blip-vqa docstring (#72) * Refine flamingo docstring (#73) * [Feature]: BLIP2 VQA (#61) * [Feature]: VQA forward * [Feature]: Reproduce accuracy * [Fix]: Fix lint * [Fix]: Add blank line * minor fix --------- Co-authored-by: yingfhu <yingfhu@gmail.com> * [Feature]: BLIP2 docstring (#74) * [Feature]: Add caption docstring * [Feature]: Add docstring to blip2 vqa * [Feature]: Add docstring to retrieval * Update BLIP-2 metafile and README (#75) * [Feature]: Add readme and docstring * Update blip2 results --------- Co-authored-by: mzr1996 <mzr1996@163.com> * [Feature] BLIP Visual Grounding on MMPretrain Branch (#66) * blip grounding merge with mmpretrain * remove commit * blip grounding test and inference api * refcoco dataset * refcoco dataset refine config * rebasing * gitignore * rebasing * minor edit * minor edit * Update blip-vqa docstring (#72) * rebasing * Revert "minor edit" This reverts commit 639cec757c215e654625ed0979319e60f0be9044. * blip grounding final * precommit * refine config * refine config * Update blip visual grounding --------- Co-authored-by: Yiqin Wang 王逸钦 <wyq1217@outlook.com> Co-authored-by: mzr1996 <mzr1996@163.com> * Update visual grounding metric * Update OFA docstring, README and metafiles. (#76) * [Docs] Update installation docs and gradio demo docs. (#77) * Update OFA name * Update Visual Grounding Visualizer * Integrate accelerate support * Fix imports. * Fix timm backbone * Update imports * Update README * Update circle ci * Update flamingo config * Add gradio demo README * [Feature]: Add scienceqa (#1571) * [Feature]: Add scienceqa * [Feature]: Change param name * Update docs * Update video --------- Co-authored-by: Hubert <42952108+yingfhu@users.noreply.github.com> Co-authored-by: yingfhu <yingfhu@gmail.com> Co-authored-by: Yuan Liu <30762564+YuanLiuuuuuu@users.noreply.github.com> Co-authored-by: Yiqin Wang 王逸钦 <wyq1217@outlook.com> Co-authored-by: Rongjie Li <limo97@163.com>
2023-05-19 16:50:04 +08:00
- configs/blip/metafile.yml
- configs/flamingo/metafile.yml
- configs/blip2/metafile.yml
[Feature] Support Chinese CLIP. (#1576) * support cn-clip * update README * Update progress bar * update order of category * fix lint * update * update readme and metafile * update * update docstring * refactor tokenizer * fix lint * Update README and progress bar --------- Co-authored-by: mzr1996 <mzr1996@163.com>
2023-05-22 15:46:13 +08:00
- configs/chinese_clip/metafile.yml
[Feature] Support iTPN and HiViT (#1584) * hivit added * Update hivit.py * Update hivit.py * Add files via upload * Update __init__.py * Add files via upload * Update __init__.py * Add files via upload * Update hivit.py * Add files via upload * Add files via upload * Add files via upload * Add files via upload * Update itpn.py * Add files via upload * Update __init__.py * Update mae_hivit-base-p16.py * Delete mim_itpn-base-p16.py * Add files via upload * Update itpn_hivit-base-p16.py * Update itpn.py * Update hivit.py * Update __init__.py * Update mae.py * Delete hivit.py * Update __init__.py * Delete configs/itpn directory * Add files via upload * Add files via upload * Delete configs/hivit directory * Add files via upload * refactor and add metafile and readme * update clip * add ut * update ut * update * update docstring * update model.rst --------- Co-authored-by: 田运杰 <48153283+sunsmarterjie@users.noreply.github.com>
2023-05-26 12:08:34 +08:00
- configs/itpn/metafile.yml
- configs/hivit/metafile.yml
[Feature] Support SparK. (#1531) * add spark configs * fix configs * remove repeat aug * add module codes * support lr layer decay of resnet * update * fix lint * add metafile and readme * fix lint * add models and logs * refactor codes * fix lint * update model rst * update name * add docstring * add ut * fix lint --------- Co-authored-by: Ma Zerun <mzr1996@163.com>
2023-06-19 11:27:50 +08:00
- configs/spark/metafile.yml
[Feature] Support MiniGPT-4 (#1642) * support inference of MiniGPT-4 * refine codes * update metafile, readme and docs * fix typo * fix lint * add ckpt load hook
2023-06-16 22:50:34 +08:00
- configs/minigpt4/metafile.yml
[Feature] Support LLaVA (#1652)
2023-06-17 16:05:52 +08:00
- configs/llava/metafile.yml
[Feature] Support otter (#1651) * [Feature] Support Otter * Update docs
2023-06-17 16:03:21 +08:00
- configs/otter/metafile.yml
[Feature]: Add MFF (#1725) * [Feature]: Add MFF * [Feature]: Add mff linear prob * [Feature]: Add ft * [Fix]: Update docstring * [Feature]: Update out_indices * [Feature]: Add prefix to ft * [Feature]: Add README * [Feature]: Update readme * [Feature]: Update README * [Feature]: Add metafile * [Feature]: Update README * [Fix]: Fix lint * [Feature]: Add UT * [Feature]: Update paper link
2023-08-08 16:01:07 +08:00
- configs/mff/metafile.yml