From 93e5ac27d8d8e44e9ad68c2cc18283de514ae4b8 Mon Sep 17 00:00:00 2001 From: anijain2305 Date: Wed, 29 Apr 2020 22:32:33 +0000 Subject: [PATCH] Removing older Object detection TFlite test --- tests/python/frontend/tflite/test_forward.py | 20 -------------------- 1 file changed, 20 deletions(-) diff --git a/tests/python/frontend/tflite/test_forward.py b/tests/python/frontend/tflite/test_forward.py index bc3f32a2b0cd..ae2a4c6b5f3d 100644 --- a/tests/python/frontend/tflite/test_forward.py +++ b/tests/python/frontend/tflite/test_forward.py @@ -1929,25 +1929,6 @@ def test_forward_qnn_mobilenet_v3_net(): tvm_sorted_labels = tvm_predictions.argsort()[-3:][::-1] tvm.testing.assert_allclose(tvm_sorted_labels, tflite_sorted_labels) -####################################################################### -# SSD Mobilenet -# ------------- - -def test_forward_ssd_mobilenet_v1(): - """Test the SSD Mobilenet V1 TF Lite model.""" - # SSD MobilenetV1 - tflite_model_file = tf_testing.get_workload_official( - "https://raw.githubusercontent.com/dmlc/web-data/master/tensorflow/models/object_detection/ssd_mobilenet_v1_coco_2018_01_28_nopp.tgz", - "ssd_mobilenet_v1_coco_2018_01_28_nopp.tflite") - with open(tflite_model_file, "rb") as f: - tflite_model_buf = f.read() - np.random.seed(0) - data = np.random.uniform(size=(1, 300, 300, 3)).astype('float32') - tflite_output = run_tflite_graph(tflite_model_buf, data) - tvm_output = run_tvm_graph(tflite_model_buf, data, 'normalized_input_image_tensor', num_output=2) - for i in range(2): - tvm.testing.assert_allclose(np.squeeze(tvm_output[i]), np.squeeze(tflite_output[i]), - rtol=1e-5, atol=2e-5) ####################################################################### # MediaPipe @@ -2045,7 +2026,6 @@ def test_forward_mediapipe_hand_landmark(): test_forward_mobilenet_v3() test_forward_inception_v3_net() test_forward_inception_v4_net() - test_forward_ssd_mobilenet_v1() test_forward_mediapipe_hand_landmark() # End to End quantized