diff --git a/demo/models/__init__.py b/demo/models/__init__.py index ca9a16790e211b16e4e3f9993894a2d5ad1a3918..09f0d1579486ed866517055a1019b35043441f90 100644 --- a/demo/models/__init__.py +++ b/demo/models/__init__.py @@ -1,13 +1,14 @@ from __future__ import absolute_import from .mobilenet import MobileNet from .resnet import ResNet34, ResNet50 -from .resnet_vd import ResNet50_vd -from .mobilenet_v2 import MobileNetV2 +from .resnet_vd import ResNet50_vd, ResNet101_vd +from .mobilenet_v2 import MobileNetV2_x0_25, MobileNetV2 from .pvanet import PVANet __all__ = [ "model_list", "MobileNet", "ResNet34", "ResNet50", "MobileNetV2", "PVANet", - "ResNet50_vd" + "ResNet50_vd", "ResNet101_vd", "MobileNetV2_x0_25" ] model_list = [ - 'MobileNet', 'ResNet34', 'ResNet50', 'MobileNetV2', 'PVANet', 'ResNet50_vd' + 'MobileNet', 'ResNet34', 'ResNet50', 'MobileNetV2', 'PVANet', + 'ResNet50_vd', "ResNet101_vd", "MobileNetV2_x0_25" ]