mmpretrain/tests/test_models/test_backbones
Yixiao Fang 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
..
__init__.py [Enhance] Support dynamic input shape for ViT-based algorithms. (#706) 2022-03-03 13:10:12 +08:00
test_beit.py [Refactor] Refactor BEiT backbone and support v1/v2 inference. (#1144) 2022-11-29 12:56: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 Activation Checkpointing for ConvNeXt. (#1153) 2022-11-14 15:04:28 +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 [Enhance] Upgrade isort pre-commit hooks. (#687) 2022-02-17 02:17:20 +08:00
test_hornet.py [Feature] Support HorNet Backbone for dev1.x. (#1094) 2022-11-04 15:33:46 +08:00
test_hrnet.py [Feature] Support HRNet and add pre-trained models (#660) 2022-01-28 10:54:14 +08:00
test_inception_v3.py [Refactor] Refactor dev scripts 2022-07-18 11:11:13 +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 [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_mobilenet_v3.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
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 [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
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 [Docs] Add Copyright information. (#413) 2021-08-17 19:52:42 +08:00
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 [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_seresnet.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_seresnext.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_shufflenet_v1.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
test_shufflenet_v2.py [Enhance] Improve downstream repositories compatibility (#421) 2021-09-08 10:38:57 +08:00
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_tnt.py [Refactor] Fix TnT compatibility and verbose warning. (#436) 2021-09-22 19:37:24 +08:00
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 [Refactor] Use `mmengine` instead of `mmcv` and refactor some transforms. (#986) 2022-08-24 15:59:02 +08:00
utils.py [Enhance] Support dynamic input shape for ViT-based algorithms. (#706) 2022-03-03 13:10:12 +08:00