Skip to content

T
tic
Commit 3a133550 by Alexander Pivovarov Committed by Tianqi Chen
Options

Compare all outputs in TFLite test_forward_ssd_mobilenet_v1 (#4373)

parent 1df8d308
Showing with 4 additions and 3 deletions
tests/python/frontend/tflite/test_forward.py
...@@ -1134,9 +1134,10 @@ def test_forward_ssd_mobilenet_v1(): ...@@ -1134,9 +1134,10 @@ def test_forward_ssd_mobilenet_v1():
tflite_model_buf = f.read() tflite_model_buf = f.read()
data = np.random.uniform(size=(1, 300, 300, 3)).astype('float32') data = np.random.uniform(size=(1, 300, 300, 3)).astype('float32')
tflite_output = run_tflite_graph(tflite_model_buf, data) tflite_output = run_tflite_graph(tflite_model_buf, data)
tvm_output = run_tvm_graph(tflite_model_buf, data, 'normalized_input_image_tensor') tvm_output = run_tvm_graph(tflite_model_buf, data, 'normalized_input_image_tensor', num_output=2)
tvm.testing.assert_allclose(np.squeeze(tvm_output[0]), np.squeeze(tflite_output[0]), for i in range(2):
rtol=1e-5, atol=1e-5) tvm.testing.assert_allclose(np.squeeze(tvm_output[i]), np.squeeze(tflite_output[i]),
rtol=1e-5, atol=2e-5)
####################################################################### #######################################################################
# MediaPipe # MediaPipe
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment