mmclassification/tools/model_converters
aso538 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
..
clip_to_mmcls.py [Feature] Add clip backbone. (#1258) 2022-12-22 16:33:57 +08:00
convnext_to_mmcls.py [Enhance] Support ConvNeXt More Weights. (#1240) 2022-12-19 17:17:57 +08:00
davit_to_mmcls.py [Feature] Support DaViT. (#1105) 2022-11-16 17:23:55 +08:00
deit3_to_mmcls.py [Feature] Support DeiT3. (#1065) 2022-10-10 14:54:20 +08:00
edgenext_to_mmcls.py
efficientnet_to_mmcls.py [Feature] EfficientNets NoisyStudent & L2. (#1122) 2022-11-21 16:57:13 +08:00
efficientnetv2_to_mmcls.py [Feature] [CodeCamp #68] Add EfficientnetV2 Backbone. (#1253) 2022-12-30 15:18:39 +08:00
eva_to_mmcls.py [Feature] Support EVA. (#1239) 2022-12-14 13:21:33 +08:00
hornet2mmcls.py [Feature] Support HorNet Backbone for dev1.x. (#1094) 2022-11-04 15:33:46 +08:00
levit2mmcls.py [Feature] Support LeViT backbone. (#1238) 2023-01-17 17:43:42 +08:00
mixmimx_to_mmcls.py [Feature] Add mixmim backbone with checkpoints. (#1224) 2022-12-20 16:52:54 +08:00
mlpmixer_to_mmcls.py
mobilenetv2_to_mmcls.py
publish_model.py
reparameterize_model.py [Bug] Fix `reparameterize_model.py` doesn't save meta info. (#1221) 2022-12-06 17:06:31 +08:00
replknet_to_mmcls.py [Feature] Support RepLKnet backbone. (#1129) 2022-11-21 10:18:58 +08:00
repvgg_to_mmcls.py
revvit_to_mmcls.py [Feature] Implementation of RevViT. (#1127) 2022-12-14 11:46:39 +08:00
shufflenetv2_to_mmcls.py
tinyvit_to_mmcls.py [Feature] Add TinyViT for dev-1.x. (#1042) 2022-12-20 13:04:00 +08:00
torchvision_to_mmcls.py
twins2mmcls.py
van2mmcls.py
vgg_to_mmcls.py
vig_to_mmcls.py [Feature] Support VIG Backbone. (#1304) 2023-01-17 16:55:56 +08:00