__init__.py 420 字节
Newer Older
1
from __future__ import absolute_import
2 3
from .mobilenet import MobileNet
from .resnet import ResNet34, ResNet50
B
Bai Yifan 已提交
4
from .resnet_vd import ResNet50_vd
5
from .mobilenet_v2 import MobileNetV2
W
whs 已提交
6
from .pvanet import PVANet
B
Bai Yifan 已提交
7
__all__ = [
8 9 10 11
    "model_list", "MobileNet", "ResNet34", "ResNet50", "MobileNetV2", "PVANet",
    "ResNet50_vd"
]
model_list = [
B
Bai Yifan 已提交
12 13
    'MobileNet', 'ResNet34', 'ResNet50', 'MobileNetV2', 'PVANet', 'ResNet50_vd'
]