From ea1e00a4315f958b285093431efa13cfd41b583f Mon Sep 17 00:00:00 2001 From: Leandro Nunes Date: Mon, 15 Jun 2020 10:06:28 +0100 Subject: [PATCH] Pin hand landmark network to version 0.7.4. * Versions above 0.7.4 are broken due to changes in the quantization operations in the model, which are current not supported by TVM. Fixes #5774. --- tests/python/frontend/tflite/test_forward.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/python/frontend/tflite/test_forward.py b/tests/python/frontend/tflite/test_forward.py index 5b2f85094a27..1ce4997aa73b 100644 --- a/tests/python/frontend/tflite/test_forward.py +++ b/tests/python/frontend/tflite/test_forward.py @@ -2437,12 +2437,11 @@ def test_forward_coco_ssd_mobilenet_v1(): ####################################################################### # MediaPipe # ------------- -@pytest.mark.skip("TFLite problem") def test_forward_mediapipe_hand_landmark(): """Test MediaPipe 2D hand landmark TF Lite model.""" # MediaPipe 2D hand landmark TF tflite_model_file = download_testdata( - "https://github.com/google/mediapipe/raw/master/mediapipe/models/hand_landmark.tflite", + "https://github.com/google/mediapipe/raw/v0.7.4/mediapipe/models/hand_landmark.tflite", "hand_landmark.tflite") with open(tflite_model_file, "rb") as f: tflite_model_buf = f.read()