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

Updated the no_face_detected test case

Browse Source
This commit is contained in:
kinaryml 2023-04-14 11:27:27 -07:00
parent e17d401c6e
commit c3a1b1add7
1 changed files with 1 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -180,9 +180,7 @@ class FaceStylizerTest(parameterized.TestCase):
image_processing_options = _ImageProcessingOptions(roi)
# Performs face stylization on the input.
stylized_image = stylizer.stylize(test_image, image_processing_options)
self.assertIsInstance(stylized_image, _Image)
self.assertEqual(stylized_image.width, _MODEL_IMAGE_SIZE)
self.assertEqual(stylized_image.height, _MODEL_IMAGE_SIZE)
self.assertIsNone(stylized_image)
def test_missing_result_callback(self):
options = _FaceStylizerOptions(