034919d032
[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
0826df8963
[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
1ee9bbe050
[Docs] Update links ( #1457 )
...
* update links
* update readtherdocs
* update
* update
* fix lint
* update
* update
* update
* update cov branch
* update
* update
* update
2023-04-06 20:58:52 +08:00
f6b65fcbe7
[Docs] Update get start docs and user guides. ( #1407 )
...
* update user_guides
* update test.md
* fix lint
* fix typo
* refine
* fix typo
* update retriever to api
* update rst and downstream
* update index.rst
* update index.rst
* update custom.js
* update chinese docs
* update config.md
* update train and test
* add pretrain on custom dataset
* fix lint
2023-03-20 15:56:09 +08:00
08dc8c75d3
[Refactor] Add selfsup algorithms. ( #1389 )
...
* remove basehead
* add moco series
* add byol simclr simsiam
* add ut
* update configs
* add simsiam hook
* add and refactor beit
* update ut
* add cae
* update extract_feat
* refactor cae
* add mae
* refactor data preprocessor
* update heads
* add maskfeat
* add milan
* add simmim
* add mixmim
* fix lint
* fix ut
* fix lint
* add eva
* add densecl
* add barlowtwins
* add swav
* fix lint
* update readtherdocs rst
* update docs
* update
* Decrease UT memory usage
* Fix docstring
* update DALLEEncoder
* Update model docs
* refactor dalle encoder
* update docstring
* fix ut
* fix config error
* add val_cfg and test_cfg
* refactor clip generator
* fix lint
* pass check
* fix ut
* add lars
* update type of BEiT in configs
* Use MMEngine style momentum in EMA.
* apply mmpretrain solarize
---------
Co-authored-by: mzr1996 <mzr1996@163.com>
2023-03-06 16:53:15 +08:00
dda3d6565b
[Docs] Update generate_readme.py and readme files. ( #1388 )
...
* Update generate_readme.py and readme files.
* Update reamde
* Update docs
* update metafile
* update simmim readme
* update
* update mae
* fix lint
* update mocov2
* update readme pic
* fix lint
* Fix mmcls download links.
* Fix Chinese docs.
* Decrease readthedocs requirements.
---------
Co-authored-by: fangyixiao18 <fangyx18@hotmail.com>
2023-03-02 13:29:07 +08:00
8352951f3d
[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
a3f2effb17
[Feature] Add `ImageClassificationInferencer`. ( #1261 )
...
* [Feature] Add ImageClassificationInferencer.
* Update inferencer implementation and add unit tests.
* Update documentations.
* Update pre-commit hook
* Update docs
2023-02-08 14:30:12 +08:00
6b9e2b55dd
[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
c98dc4555c
[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
74743ef588
[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
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
d80ec5a4b8
[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
b0007812d6
[Enhance] Enhance ArcFaceClsHead. ( #1181 )
...
* update arcface
* fix unit tests
* add adv-margins
add adv-margins
update arcface
* rebase
* update doc and fix ut
* rebase
* update code
* rebase
* use label data
* update set-margins
* Modify Arcface related method names.
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-11-21 18:10:39 +08:00
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
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
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
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
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
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
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
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
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
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
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
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
ccece7e6dc
[Docs] Refactor dataset tutorial ( #916 )
...
* refactor dataset tutorials
* split into user_guide and advance_guide
* refine
* Fix dataset preparasion tutorial.
* refine CN docs
* update docs API doc link
* refine new a dataset
* refine new a dataset
* refine new a dataset
Co-authored-by: mzr1996 <mzr1996@163.com>
2022-08-30 18:45:58 +08:00
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
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
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
702c196514
[Docs] Refine API reference. ( #774 )
...
* [Docs] Refine API reference
* Add PoolFormer
2022-04-15 20:19:20 +08:00
zzc98
Yixiao Fang
Ma Zerun
QINGTIAN
aso538
szwlh-c
Songyang Zhang
Ezra-Yu
Hubert
Jiahao Wang
takuoko