From ef60db0287711cb284dc25f89201772ebcfbb318 Mon Sep 17 00:00:00 2001 From: Bai Yifan Date: Tue, 12 May 2020 19:42:23 +0800 Subject: [PATCH] add ResNet101_vd and MobileNetV2_x0_25 in demo/models (#274) --- demo/models/__init__.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/demo/models/__init__.py b/demo/models/__init__.py index ca9a1679..09f0d157 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" ] -- GitLab