mirror of
https://github.com/open-mmlab/mmclassification.git
synced 2025-06-03 21:53:55 +08:00
bc1b08ba41
* add vgg * add vgg model coversion tool * fix out_indices and docstr * add vgg models in configs * add params, flops and accuracy in docs * add pretrained models url * use ConvModule and refine var names * update vgg conversion tool * modify bn config * add docs for arch_setting * add unit test for vgg * rm debug code * update vgg pretrained models
21 lines
670 B
Python
21 lines
670 B
Python
from .alexnet import AlexNet
|
|
from .lenet import LeNet5
|
|
from .mobilenet_v2 import MobileNetV2
|
|
from .mobilenet_v3 import MobileNetv3
|
|
from .regnet import RegNet
|
|
from .resnest import ResNeSt
|
|
from .resnet import ResNet, ResNetV1d
|
|
from .resnet_cifar import ResNet_CIFAR
|
|
from .resnext import ResNeXt
|
|
from .seresnet import SEResNet
|
|
from .seresnext import SEResNeXt
|
|
from .shufflenet_v1 import ShuffleNetV1
|
|
from .shufflenet_v2 import ShuffleNetV2
|
|
from .vgg import VGG
|
|
|
|
__all__ = [
|
|
'LeNet5', 'AlexNet', 'VGG', 'RegNet', 'ResNet', 'ResNeXt', 'ResNetV1d',
|
|
'ResNeSt', 'ResNet_CIFAR', 'SEResNet', 'SEResNeXt', 'ShuffleNetV1',
|
|
'ShuffleNetV2', 'MobileNetV2', 'MobileNetv3'
|
|
]
|