cassdlcm/mediapipe
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Added Face Stylizer tests for the Python Tasks API

Browse Source
This commit is contained in:
kinaryml 2023-04-20 17:54:59 -07:00
parent a5716c9225
commit 966ca43f99
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mediapipe/tasks/python/test/vision/face_stylizer_test.py
View File

@ -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.