MNISTのテストを追加

sony · Jun 1, 2018 · c94a11a · c94a11a
1 parent 7882884
commit c94a11a
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/python/test/utils/conversion/test_conversion.py b/python/test/utils/conversion/test_conversion.py
@@ -1270,6 +1270,18 @@ def test_nnp_onnx_conversion_bvlc_caffenet(tmpdir, nnp_fixture):
                                     "bvlc_caffenet.onnx",
                                     "prob_1", "exec_0",
                                     in_name="data_0", in_img=img)
+
+
+#@pytest.mark.slow
+#def test_onnx_nnp_conversion_mnist(tmpdir, nnp_fixture):
+#    img = np.random.rand(1, 1, 28, 28).astype(np.float32)
+#    convert_onnx_to_nnp_and_compare(tmpdir, TEST_DATA_DIR,
+#                                    "mnist.onnx", "mnist.nnp",
+#                                    "Plus214_Output_0", "exec_0",
+#                                    in_name="Input3", in_img=img,
+#                                    backend="cntk",
+#                                    export_nnp_path=TEST_DATA_DIR)
+
 #@pytest.mark.slow
 #def test_onnx_nnp_conversion_inception_v1(tmpdir, nnp_fixture):
 #    img = np.random.rand(1, 3, 224, 224).astype(np.float32)