Skip to content

Commit

Permalink
bvlc_googlenetの単体テストを追加
Browse files Browse the repository at this point in the history
ONNX => NNPは1e-8まで一致、NNP => ONNXは1e-5まで一致
  • Loading branch information
Masato Hori committed Jun 1, 2018
1 parent 09b3d74 commit f4ed404
Showing 1 changed file with 21 additions and 8 deletions.
29 changes: 21 additions & 8 deletions python/test/utils/conversion/test_conversion.py
Original file line number Diff line number Diff line change
Expand Up @@ -1233,14 +1233,27 @@ def test_nnp_onnx_conversion_zfnet512(tmpdir, nnp_fixture):
in_name="gpu_0/data_0", in_img=img)


#@pytest.mark.slow
#def test_onnx_nnp_conversion_bvlc_googlenet(tmpdir, nnp_fixture):
# img = np.random.rand(1, 3, 224, 224).astype(np.float32)
# convert_onnx_to_nnp_and_compare(tmpdir, TEST_DATA_DIR,
# "bvlc_googlenet.onnx", "bvlc_googlenet.nnp",
# "prob_1", "exec_0",
# in_name="data_0", in_img=img)
#
@pytest.mark.slow
def test_onnx_nnp_conversion_bvlc_googlenet(tmpdir, nnp_fixture):
img = np.random.rand(1, 3, 224, 224).astype(np.float32)
convert_onnx_to_nnp_and_compare(tmpdir, TEST_DATA_DIR,
"bvlc_googlenet.onnx",
"bvlc_googlenet.nnp",
"prob_1", "exec_0",
in_name="data_0", in_img=img)


@pytest.mark.slow
def test_nnp_onnx_conversion_bvlc_googlenet(tmpdir, nnp_fixture):
img = np.random.rand(1, 3, 224, 224).astype(np.float32)
convert_nnp_to_onnx_and_compare(tmpdir, TEST_DATA_DIR,
"bvlc_googlenet.nnp",
"bvlc_googlenet.onnx",
"prob_1", "exec_0",
in_name="data_0", in_img=img,
atol=1e-5)


#@pytest.mark.slow
#def test_onnx_nnp_conversion_bvlc_caffenet(tmpdir, nnp_fixture):
# img = np.random.rand(1, 3, 224, 224).astype(np.float32)
Expand Down

0 comments on commit f4ed404

Please sign in to comment.