Ezra-Yu 922a31a68e

[Feature] ADD RepMLP Backbone and checkpoints. (#709) ...

* add repmlp

* refactor backbone code

* add checkpoint, readme, unit tests

* unchange demo

* improve docstring.

* update pic

* refactor patch embed

* refactor unit tests

* fix lint

* update tools

* update tools

* Update checkpoint path

Co-authored-by: mzr1996 <mzr1996@163.com>

2022-03-30 20:40:25 +08:00

..

__init__.py

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

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

test_conformer.py

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

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

test_convmixer.py

[Feature] Support ConvMixer. (#716)

2022-03-23 15:23:57 +08:00

test_convnext.py

[Feature] Support ConvNeXt (#670)

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

test_cspnet.py

[Feature] Add CSPNet and backbone and checkpoints (#735)

2022-03-30 15:29:35 +08:00

test_deit.py

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

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

test_efficientnet.py

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

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

test_hrnet.py

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

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

test_mlp_mixer.py

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

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

test_mobilenet_v2.py

…

test_mobilenet_v3.py

…

test_regnet.py

…

test_repmlp.py

[Feature] ADD RepMLP Backbone and checkpoints. (#709)

2022-03-30 20:40:25 +08:00

test_repvgg.py

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

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

test_res2net.py

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

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

test_resnest.py

…

test_resnet_cifar.py

…

test_resnet.py

[Feature] Add ResNetV1c. (#692)

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

test_resnext.py

…

test_seresnet.py

…

test_seresnext.py

…

test_shufflenet_v1.py

…

test_shufflenet_v2.py

…

test_swin_transformer.py

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

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

test_t2t_vit.py

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

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

test_timm_backbone.py

[Feature] Support features_only in TIMMBackbone (#668)

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

test_tnt.py

[Refactor] Fix TnT compatibility and verbose warning. (#436)

2021-09-22 19:37:24 +08:00

test_twins.py

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

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

test_vgg.py

…

test_vision_transformer.py

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

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

utils.py

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

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