mirrors
/
mmclassification
mirror of https://github.com/open-mmlab/mmclassification.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
mmclassification/tests/test_models/test_backbones
History
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
..
__init__.py
…
test_beit.py [Feature] Support EVA. (#1239) 2022-12-14 13:21:33 +08:00
test_conformer.py [CI] Update circle-ci and github workflow. (#1018) 2022-09-13 11:46:15 +08:00
test_convmixer.py [CI] Update circle-ci and github workflow. (#1018) 2022-09-13 11:46:15 +08:00
test_convnext.py [Feature] Support ConvNeXt-V2 backbone. (#1294) 2023-01-06 16:13:41 +08:00
test_cspnet.py [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00
test_davit.py [Feature] Support DaViT. (#1105) 2022-11-16 17:23:55 +08:00
test_deit.py [Refactor] Use `mmengine` instead of `mmcv` and refactor some transforms. (#986) 2022-08-24 15:59:02 +08:00
test_deit3.py [Feature] Support DeiT3. (#1065) 2022-10-10 14:54:20 +08:00
test_densenet.py [Feature] Support DenseNet. (#750) 2022-04-28 13:06:03 +08:00
test_edgenext.py [Feature] Support EdgeNeXt for dev-1.x. (#1037) 2022-09-21 13:27:04 +08:00
test_efficientformer.py [Feature] Add efficientformer Backbone for MMCls 1.x. (#1031) 2022-09-20 14:56:45 +08:00
test_efficientnet.py
…
test_efficientnet_v2.py [Feature] [CodeCamp #68] Add EfficientnetV2 Backbone. (#1253) 2022-12-30 15:18:39 +08:00
test_hornet.py [Feature] Support HorNet Backbone for dev1.x. (#1094) 2022-11-04 15:33:46 +08:00
test_hrnet.py
…
test_inception_v3.py [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +08:00
test_levit.py [Feature] Support LeViT backbone. (#1238) 2023-01-17 17:43:42 +08:00
test_mixmim.py [Feature] Add mixmim backbone with checkpoints. (#1224) 2022-12-20 16:52:54 +08:00
test_mlp_mixer.py [Improve] Clean useless code and reduce unit tests memory usage. 2022-07-18 11:11:13 +08:00
test_mobilenet_v2.py
…
test_mobilenet_v3.py
…
test_mobileone.py [Feature] Add MobileOne Backbone For MMCls 1.x. (#1030) 2022-09-16 11:55:10 +08:00
test_mobilevit.py [Feature] Support MobileViT backbone. (#1068) 2022-10-18 17:05:59 +08:00
test_mvit.py [Feature] Support MViT for MMCLS 1.x (#1023) 2022-09-22 09:23:49 +08:00
test_poolformer.py [Feature] Add PoolFormer backbone and checkpoints. (#746) 2022-04-15 11:35:15 +08:00
test_regnet.py
…
test_replknet.py [Feature] Support RepLKnet backbone. (#1129) 2022-11-21 10:18:58 +08:00
test_repmlp.py [Refactor] Use `mmengine` instead of `mmcv` and refactor some transforms. (#986) 2022-08-24 15:59:02 +08:00
test_repvgg.py [Feature] Support RepLKnet backbone. (#1129) 2022-11-21 10:18:58 +08:00
test_res2net.py [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00
test_resnest.py
…
test_resnet.py [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00
test_resnet_cifar.py [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00
test_resnext.py
…
test_revvit.py [Feature] Implementation of RevViT. (#1127) 2022-12-14 11:46:39 +08:00
test_seresnet.py
…
test_seresnext.py
…
test_shufflenet_v1.py
…
test_shufflenet_v2.py
…
test_swin_transformer.py [Fix] Fix the output position of Swin-Transformer. (#947) 2022-09-22 17:11:52 +08:00
test_swin_transformer_v2.py [Feature] Support Swin Transformer V2 for MMCLS 1.x. (#1029) 2022-09-20 15:45:27 +08:00
test_t2t_vit.py [Refactor] Use `mmengine` instead of `mmcv` and refactor some transforms. (#986) 2022-08-24 15:59:02 +08:00
test_timm_backbone.py [Feature] Add TIMM and HuggingFace wrappers to build classifiers from them directly. (#1102) 2022-11-10 14:56:19 +08:00
test_tinyvit.py [Feature] Add TinyViT for dev-1.x. (#1042) 2022-12-20 13:04:00 +08:00
test_tnt.py
…
test_twins.py [Refactor] Add VOC dataset and mutil_label dataset 2022-07-18 11:11:13 +08:00
test_van.py [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00
test_vgg.py [Refactor] Refactor the data flow. (#989) 2022-08-26 10:40:43 +08:00
test_vision_transformer.py [Feature] Add clip backbone. (#1258) 2022-12-22 16:33:57 +08:00
utils.py
…