mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
74743ef588
* add efficientnet_v2.py * add efficientnet_v2 in __init__.py * add efficientnet_v2_s base config file * add efficientnet_v2 config file * add efficientnet_v2 config file * update tuple output * update config file * update model file * update model file * update model file * update config file * update model file * update config file * update model file * update model file * update model file * update model file * update model file * update config file * update config file * update model file * update model file * update model file * update model file * update model config file * Update efficientnet_v2.py * add config file and modify arch * add config file and modify arch * add the file about convert_pth from timm to mmcls * update efficientnetv2 model file with mmcls style * add the file about convert_pth from timm to mmcls * add the file about convert_pth from timm to mmcls * update convert file * update model file * update convert file * update model file * update model file * update model file * add metefile and README * Update tools/model_converters/efficientnetv2-timm_to_mmcls.py Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com> * update model file and convert file * update model file and convert file * update model file and convert file * update model file and convert file * update model file * update model file * update model file * update config file and README file * update metafile * Update efficientnetv2_to_mmcls.py * update model-index.yml * update metafile.yml * update b0 and s train pipeline * update b0 and s train pipeline * update b0 and s train pipeline * add test_efficientnet_v2 * update test_efficientnet_v2 * update model file docs * update test_efficientnet_v2 * update test_efficientnet_v2 * add efficientnet_v2.py * add efficientnet_v2 in __init__.py * add efficientnet_v2_s base config file * add efficientnet_v2 config file * add efficientnet_v2 config file * update tuple output * update config file * update model file * update model file * update model file * update model file * update config file * update config file * update model file * update model file * update model file * update model file * update model file * update config file * update config file * update model file * update model file * update model file * update model file * update model config file * Update efficientnet_v2.py * add config file and modify arch * add config file and modify arch * add the file about convert_pth from timm to mmcls * update efficientnetv2 model file with mmcls style * add the file about convert_pth from timm to mmcls * add the file about convert_pth from timm to mmcls * update convert file * update model file * update convert file * update model file * update model file * update model file * add metefile and README * Update tools/model_converters/efficientnetv2-timm_to_mmcls.py Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com> * update model file and convert file * update model file and convert file * update model file and convert file * update model file and convert file * update model file * update model file * update model file * update config file and README file * update metafile * Update efficientnetv2_to_mmcls.py * update model-index.yml * update metafile.yml * update b0 and s train pipeline * update b0 and s train pipeline * update b0 and s train pipeline * add test_efficientnet_v2 * update test_efficientnet_v2 * update model file docs * update test_efficientnet_v2 * update test_efficientnet_v2 * pass pre-commit hook * refactor efficientnetv2 * refactor efficientnetv2 * update readme, metafile and weight links * update model-index.yml * fix lint * fix typo * Update efficientnetv2-b1_8xb32_in1k.py * Update efficientnetv2-b2_8xb32_in1k.py * Update efficientnetv2-b3_8xb32_in1k.py * update two moduals and model file * update modual file * update accuracys * update accuracys * update metafile * fix build docs * update links * update README.md Co-authored-by: qingtian <459291290@qq.com> Co-authored-by: Ezra-Yu <18586273+Ezra-Yu@users.noreply.github.com>
111 lines
2.7 KiB
Python
111 lines
2.7 KiB
Python
# Copyright (c) OpenMMLab. All rights reserved.
|
|
from .alexnet import AlexNet
|
|
from .beit import BEiT
|
|
from .conformer import Conformer
|
|
from .convmixer import ConvMixer
|
|
from .convnext import ConvNeXt
|
|
from .cspnet import CSPDarkNet, CSPNet, CSPResNet, CSPResNeXt
|
|
from .davit import DaViT
|
|
from .deit import DistilledVisionTransformer
|
|
from .deit3 import DeiT3
|
|
from .densenet import DenseNet
|
|
from .edgenext import EdgeNeXt
|
|
from .efficientformer import EfficientFormer
|
|
from .efficientnet import EfficientNet
|
|
from .efficientnet_v2 import EfficientNetV2
|
|
from .hornet import HorNet
|
|
from .hrnet import HRNet
|
|
from .inception_v3 import InceptionV3
|
|
from .lenet import LeNet5
|
|
from .mixmim import MixMIMTransformer
|
|
from .mlp_mixer import MlpMixer
|
|
from .mobilenet_v2 import MobileNetV2
|
|
from .mobilenet_v3 import MobileNetV3
|
|
from .mobileone import MobileOne
|
|
from .mobilevit import MobileViT
|
|
from .mvit import MViT
|
|
from .poolformer import PoolFormer
|
|
from .regnet import RegNet
|
|
from .replknet import RepLKNet
|
|
from .repmlp import RepMLPNet
|
|
from .repvgg import RepVGG
|
|
from .res2net import Res2Net
|
|
from .resnest import ResNeSt
|
|
from .resnet import ResNet, ResNetV1c, ResNetV1d
|
|
from .resnet_cifar import ResNet_CIFAR
|
|
from .resnext import ResNeXt
|
|
from .revvit import RevVisionTransformer
|
|
from .seresnet import SEResNet
|
|
from .seresnext import SEResNeXt
|
|
from .shufflenet_v1 import ShuffleNetV1
|
|
from .shufflenet_v2 import ShuffleNetV2
|
|
from .swin_transformer import SwinTransformer
|
|
from .swin_transformer_v2 import SwinTransformerV2
|
|
from .t2t_vit import T2T_ViT
|
|
from .timm_backbone import TIMMBackbone
|
|
from .tinyvit import TinyViT
|
|
from .tnt import TNT
|
|
from .twins import PCPVT, SVT
|
|
from .van import VAN
|
|
from .vgg import VGG
|
|
from .vision_transformer import VisionTransformer
|
|
|
|
__all__ = [
|
|
'LeNet5',
|
|
'AlexNet',
|
|
'VGG',
|
|
'RegNet',
|
|
'ResNet',
|
|
'ResNeXt',
|
|
'ResNetV1d',
|
|
'ResNeSt',
|
|
'ResNet_CIFAR',
|
|
'SEResNet',
|
|
'SEResNeXt',
|
|
'ShuffleNetV1',
|
|
'ShuffleNetV2',
|
|
'MobileNetV2',
|
|
'MobileNetV3',
|
|
'VisionTransformer',
|
|
'SwinTransformer',
|
|
'TNT',
|
|
'TIMMBackbone',
|
|
'T2T_ViT',
|
|
'Res2Net',
|
|
'RepVGG',
|
|
'Conformer',
|
|
'MlpMixer',
|
|
'DistilledVisionTransformer',
|
|
'PCPVT',
|
|
'SVT',
|
|
'EfficientNet',
|
|
'EfficientNetV2',
|
|
'ConvNeXt',
|
|
'HRNet',
|
|
'ResNetV1c',
|
|
'ConvMixer',
|
|
'EdgeNeXt',
|
|
'CSPDarkNet',
|
|
'CSPResNet',
|
|
'CSPResNeXt',
|
|
'CSPNet',
|
|
'RepLKNet',
|
|
'RepMLPNet',
|
|
'PoolFormer',
|
|
'DenseNet',
|
|
'VAN',
|
|
'InceptionV3',
|
|
'MobileOne',
|
|
'EfficientFormer',
|
|
'SwinTransformerV2',
|
|
'MViT',
|
|
'DeiT3',
|
|
'HorNet',
|
|
'MobileViT',
|
|
'DaViT',
|
|
'BEiT',
|
|
'RevVisionTransformer',
|
|
'MixMIMTransformer',
|
|
'TinyViT',
|
|
]
|