mmpretrain/mmcls/models/backbones/__init__.py

12 lines
344 B
Python
Raw Normal View History

2020-06-16 14:37:03 +08:00
from .mobilenet_v2 import MobileNetV2
2020-06-12 14:40:27 +08:00
from .resnet import ResNet, ResNetV1d
2020-06-14 12:08:37 +08:00
from .resnext import ResNeXt
2020-06-17 14:20:20 +08:00
from .seresnet import SEResNet
2020-06-16 14:37:03 +08:00
from .shufflenet_v1 import ShuffleNetV1
from .shufflenet_v2 import ShuffleNetV2
2020-06-03 15:51:17 +08:00
2020-06-15 02:16:13 +08:00
__all__ = [
2020-06-17 14:20:20 +08:00
'ResNet', 'ResNeXt', 'ResNetV1d', 'ResNetV1d', 'SEResNet', 'ShuffleNetV1',
2020-06-16 14:37:03 +08:00
'ShuffleNetV2', 'MobileNetV2'
2020-06-15 02:16:13 +08:00
]