From 966ca43f99c899ca675e038aef92036ee73cd7eb Mon Sep 17 00:00:00 2001 From: kinaryml Date: Thu, 20 Apr 2023 17:54:59 -0700 Subject: [PATCH] Added Face Stylizer tests for the Python Tasks API --- mediapipe/tasks/python/test/vision/face_stylizer_test.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/mediapipe/tasks/python/test/vision/face_stylizer_test.py b/mediapipe/tasks/python/test/vision/face_stylizer_test.py index c57618801..b11ca9a7b 100644 --- a/mediapipe/tasks/python/test/vision/face_stylizer_test.py +++ b/mediapipe/tasks/python/test/vision/face_stylizer_test.py @@ -40,7 +40,6 @@ _ImageProcessingOptions = image_processing_options_module.ImageProcessingOptions _MODEL = 'face_stylizer.task' _LARGE_FACE_IMAGE = "portrait.jpg" -_SMALL_FACE_IMAGE = "portrait_small.jpg" _MODEL_IMAGE_SIZE = 256 _TEST_DATA_DIR = 'mediapipe/tasks/testdata/vision' @@ -90,7 +89,7 @@ class FaceStylizerTest(parameterized.TestCase): @parameterized.parameters( (ModelFileType.FILE_NAME, _LARGE_FACE_IMAGE), - (ModelFileType.FILE_CONTENT, _SMALL_FACE_IMAGE) + (ModelFileType.FILE_CONTENT, _LARGE_FACE_IMAGE) ) def test_stylize(self, model_file_type, image_file_name): # Load the test image. @@ -120,7 +119,7 @@ class FaceStylizerTest(parameterized.TestCase): @parameterized.parameters( (ModelFileType.FILE_NAME, _LARGE_FACE_IMAGE), - (ModelFileType.FILE_CONTENT, _SMALL_FACE_IMAGE) + (ModelFileType.FILE_CONTENT, _LARGE_FACE_IMAGE) ) def test_stylize_in_context(self, model_file_type, image_file_name): # Load the test image.