From 7ec0d8cf3b236c4ee5bccd02197984e3ccaae4de Mon Sep 17 00:00:00 2001 From: kinaryml Date: Wed, 9 Nov 2022 11:53:30 -0800 Subject: [PATCH] Added tolerance for vector coordinate values in embeddings --- mediapipe/tasks/python/test/vision/image_embedder_test.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/mediapipe/tasks/python/test/vision/image_embedder_test.py b/mediapipe/tasks/python/test/vision/image_embedder_test.py index e9ff50ed..18fd644a 100644 --- a/mediapipe/tasks/python/test/vision/image_embedder_test.py +++ b/mediapipe/tasks/python/test/vision/image_embedder_test.py @@ -48,6 +48,9 @@ _MODEL_FILE = 'mobilenet_v3_small_100_224_embedder.tflite' _BURGER_IMAGE_FILE = 'burger.jpg' _BURGER_CROPPED_IMAGE_FILE = 'burger_crop.jpg' _TEST_DATA_DIR = 'mediapipe/tasks/testdata/vision' +# Tolerance for embedding vector coordinate values. +_EPSILON = 1e-4 +# Tolerance for cosine similarity evaluation. _SIMILARITY_TOLERANCE = 1e-6 @@ -162,7 +165,7 @@ class ImageEmbedderTest(parameterized.TestCase): # Check embedding value. self.assertAlmostEqual(image_result.embeddings[0].embedding[0], - expected_first_value) + expected_first_value, delta=_EPSILON) # Checks cosine similarity. self._check_cosine_similarity(image_result, crop_result, quantize,