From a9d2149711ce737400b491456595cd2b3a07da99 Mon Sep 17 00:00:00 2001
From: MORITA Kazutaka <morita.kazutaka@lab.ntt.co.jp>
Date: Fri, 25 May 2018 01:32:34 +0900
Subject: [PATCH] [FRONTEND][Keras] Add test for MobileNet (#513)
---
nnvm/tests/python/frontend/keras/test_forward.py | 7 +++++++
1 file changed, 7 insertions(+)
diff --git a/nnvm/tests/python/frontend/keras/test_forward.py b/nnvm/tests/python/frontend/keras/test_forward.py
index c751b6443..5a399577f 100644
--- a/nnvm/tests/python/frontend/keras/test_forward.py
+++ b/nnvm/tests/python/frontend/keras/test_forward.py
@@ -160,6 +160,12 @@ def test_forward_resnet50():
verify_keras_frontend(keras_model)
+def test_forward_mobilenet():
+ keras_model = keras.applications.mobilenet.MobileNet(include_top=True, weights=None,
+ input_shape=(224,224,3), classes=1000)
+ verify_keras_frontend(keras_model)
+
+
if __name__ == '__main__':
test_forward_elemwise_add()
test_forward_softmax()
@@ -175,3 +181,4 @@ if __name__ == '__main__':
test_forward_vgg16()
test_forward_xception()
test_forward_resnet50()
+ test_forward_mobilenet()
--
GitLab