Ma Zerun 643fb192cd

[Enhance] Enhance feature extraction function. (#593) ...

* Fix MobileNet V3 configs

* Refactor to support more powerful feature extraction.

* Add unit tests

* Fix unit test

* Imporve according to comments

* Update checkpoints path

* Fix unit tests

* Add docstring of `simple_test`

* Add docstring of `extract_feat`

* Update model zoo

2021-12-17 15:55:02 +08:00

..

conformer

[Feature] Implement the conformer backbone. (#494)

2021-12-07 14:00:17 +08:00

regnet

Add RegNetX configs (#269)

2021-06-04 14:29:13 +08:00

swin_transformer

[Enhance] Improve downstream repositories compatibility (#421)

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

mlp_mixer_base_patch16.py

[Feature] Add MLP Mixer Backbone. (#528)

2021-11-15 18:25:22 +08:00

mlp_mixer_large_patch16.py

[Enhance] Add metafile, readme and converted models for MLP-Mixer (#539)

2021-11-24 19:04:19 +08:00

mobilenet_v2_1x.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

mobilenet_v3_large_imagenet.py

[Enhance] Enhance feature extraction function. (#593)

2021-12-17 15:55:02 +08:00

mobilenet_v3_small_cifar.py

Fix Mobilenetv3 structure and add pretrained model (#291)

2021-06-27 23:19:36 +08:00

mobilenet_v3_small_imagenet.py

[Enhance] Enhance feature extraction function. (#593)

2021-12-17 15:55:02 +08:00

repvgg-A0_in1k.py

[Feature] Add RepVGG backbone and checkpoints. (#414)

2021-09-29 11:06:23 +08:00

repvgg-B3_lbs-mixup_in1k.py

[Feature] Add RepVGG backbone and checkpoints. (#414)

2021-09-29 11:06:23 +08:00

res2net50-w14-s8.py

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

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

res2net50-w26-s4.py

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

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

res2net50-w26-s6.py

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

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

res2net50-w26-s8.py

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

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

res2net50-w48-s2.py

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

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

res2net101-w26-s4.py

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

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

resnest50.py

[Feature] Add ResNeSt configs (#332)

2021-07-14 15:58:19 +08:00

resnest101.py

[Feature] Add ResNeSt configs (#332)

2021-07-14 15:58:19 +08:00

resnest200.py

[Feature] Add ResNeSt configs (#332)

2021-07-14 15:58:19 +08:00

resnest269.py

[Feature] Add ResNeSt configs (#332)

2021-07-14 15:58:19 +08:00

resnet18_cifar.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnet18.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnet34_cifar.py

Add CIFAR10 configs and models (#38)

2020-08-26 16:34:15 +08:00

resnet34.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnet50_cifar_cutmix.py

[Feature]Add augments to models/utils (#278)

2021-06-20 09:44:51 +08:00

resnet50_cifar_mixup.py

[Feature]Add augments to models/utils (#278)

2021-06-20 09:44:51 +08:00

resnet50_cifar.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnet50_cutmix.py

[Feature]Add augments to models/utils (#278)

2021-06-20 09:44:51 +08:00

resnet50_label_smooth.py

[Bug]Fix label smooth bug (#203)

2021-04-13 13:53:56 +08:00

resnet50_mixup.py

[Feature]Add augments to models/utils (#278)

2021-06-20 09:44:51 +08:00

resnet50.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnet101_cifar.py

Add CIFAR10 configs and models (#38)

2020-08-26 16:34:15 +08:00

resnet101.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnet152_cifar.py

Add CIFAR10 configs and models (#38)

2020-08-26 16:34:15 +08:00

resnet152.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnetv1d50.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnetv1d101.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnetv1d152.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnext50_32x4d.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnext101_32x4d.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnext101_32x8d.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

resnext152_32x4d.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

seresnet50.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

seresnet101.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

seresnext50_32x4d.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

seresnext101_32x4d.py

Add configs for MNIST, CIFAR10 and ImageNet

2020-07-12 00:06:57 +08:00

shufflenet_v1_1x.py

Update shufflenet config (#31)

2020-08-13 10:46:39 +08:00

shufflenet_v2_1x.py

Update shufflenet config (#31)

2020-08-13 10:46:39 +08:00

t2t-vit-t-14.py

[Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467)

2021-10-29 10:37:16 +08:00

t2t-vit-t-19.py

[Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467)

2021-10-29 10:37:16 +08:00

t2t-vit-t-24.py

[Feature] Add Tokens-to-Token ViT backbone and converted checkpoints. (#467)

2021-10-29 10:37:16 +08:00

tnt_s_patch16_224.py

[Feature] Add transformer in transformer (#339)

2021-07-30 09:16:27 +08:00

vgg11.py

Add VGG and pretained models (#27)

2020-09-29 17:49:42 +08:00

vgg11bn.py

Add VGG and pretained models (#27)

2020-09-29 17:49:42 +08:00

vgg13.py

Add VGG and pretained models (#27)

2020-09-29 17:49:42 +08:00

vgg13bn.py

Add VGG and pretained models (#27)

2020-09-29 17:49:42 +08:00

vgg16.py

Add VGG and pretained models (#27)

2020-09-29 17:49:42 +08:00

vgg16bn.py

Add VGG and pretained models (#27)

2020-09-29 17:49:42 +08:00

vgg19.py

Add VGG and pretained models (#27)

2020-09-29 17:49:42 +08:00

vgg19bn.py

Add VGG and pretained models (#27)

2020-09-29 17:49:42 +08:00

vit-base-p16.py

[Refactor] Refator ViT (Continue #295) (#395)

2021-10-18 16:07:00 +08:00

vit-base-p32.py

[Refactor] Refator ViT (Continue #295) (#395)

2021-10-18 16:07:00 +08:00

vit-large-p16.py

[Refactor] Refator ViT (Continue #295) (#395)

2021-10-18 16:07:00 +08:00

vit-large-p32.py

[Refactor] Refator ViT (Continue #295) (#395)

2021-10-18 16:07:00 +08:00