Pin hand landmark network to version 0.7.4. (#5813)
authorLeandro Nunes <leandro.nunes@arm.com>
Mon, 15 Jun 2020 14:54:38 +0000 (15:54 +0100)
committerGitHub <noreply@github.com>
Mon, 15 Jun 2020 14:54:38 +0000 (22:54 +0800)
* 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

index 5b2f850..1ce4997 100644 (file)
@@ -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()