Ma Zerun c708770b42

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706) ...

* Move `resize_pos_embed` to `mmcls.models.utils`

* Refactor Vision Transformer

* Refactor DeiT

* Refactor MLP-Mixer

* Refactor Swin-Transformer

* Remove `indexing` arg

* Support dynamic inputs for t2t_vit

* Add copyright

* Fix bugs in swin transformer

* Add `pad_small_maps` option

* Update swin transformer

* Handle `attn_mask` in checkpoints of swin

* Imporve by comments

2022-03-03 13:10:12 +08:00

..

__init__.py

[Feature] Add ResNetV1c. (#692)

2022-02-23 11:36:33 +08:00

alexnet.py

[Enhance] Improve downstream repositories compatibility (#421)

2021-09-08 10:38:57 +08:00

base_backbone.py

[Docs] Add Copyright information. (#413)

2021-08-17 19:52:42 +08:00

conformer.py

[Fix] Fix Conformer forward with irregular input size. (#686)

2022-02-28 10:17:52 +08:00

convnext.py

[Feature] Support ConvNeXt (#670)

2022-01-28 10:36:45 +08:00

deit.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

efficientnet.py

[Feature] Support EfficientNet (#649)

2022-01-25 12:14:17 +08:00

hrnet.py

[Feature] Support HRNet and add pre-trained models (#660)

2022-01-28 10:54:14 +08:00

lenet.py

[Enhance] Improve downstream repositories compatibility (#421)

2021-09-08 10:38:57 +08:00

mlp_mixer.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

mobilenet_v2.py

[Enhance] Improve downstream repositories compatibility (#421)

2021-09-08 10:38:57 +08:00

mobilenet_v3.py

[Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470)

2021-10-13 14:33:07 +08:00

regnet.py

[Fix] Fix missing import Compose and Normalize.

2021-10-28 15:21:05 +08:00

repvgg.py

[Enhance] Upgrade isort pre-commit hooks. (#687)

2022-02-17 02:17:20 +08:00

res2net.py

[Feature] Add Res2Net backbone and converted weights. (#465)

2021-10-20 16:34:22 +08:00

resnest.py

[Docs] Add Copyright information. (#413)

2021-08-17 19:52:42 +08:00

resnet_cifar.py

[Enhance] Improve downstream repositories compatibility (#421)

2021-09-08 10:38:57 +08:00

resnet.py

[Feature] Add ResNetV1c. (#692)

2022-02-23 11:36:33 +08:00

resnext.py

[Docs] Add Copyright information. (#413)

2021-08-17 19:52:42 +08:00

seresnet.py

[Docs] Add Copyright information. (#413)

2021-08-17 19:52:42 +08:00

seresnext.py

[Docs] Add Copyright information. (#413)

2021-08-17 19:52:42 +08:00

shufflenet_v1.py

[Enhance] Improve downstream repositories compatibility (#421)

2021-09-08 10:38:57 +08:00

shufflenet_v2.py

[Enhance] Rewrite channel split operation in ShufflenetV2 (#632)

2022-01-25 14:45:28 +08:00

swin_transformer.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

t2t_vit.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00

timm_backbone.py

[Feature] Support features_only in TIMMBackbone (#668)

2022-01-25 11:22:23 +08:00

tnt.py

[Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470)

2021-10-13 14:33:07 +08:00

twins.py

[Feature] Add Twins backbone and convert checkpoints. (#642)

2022-01-27 10:25:05 +08:00

vgg.py

[Docs] Add code-spell pre-commit hook and fix a large mount of typos. (#470)

2021-10-13 14:33:07 +08:00

vision_transformer.py

[Enhance] Support dynamic input shape for ViT-based algorithms. (#706)

2022-03-03 13:10:12 +08:00