talksik/mediapipe
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4263 from kinaryml:face-stylizer-python-tests

Browse Source 
PiperOrigin-RevId: 523152290
This commit is contained in:
Copybara-Service
2023-04-10 10:39:50 -07:00
parent c036b9f408 18d5beb598
commit b685f53c6b
1 changed files with 6 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
mediapipe/tasks/python/vision/face_stylizer.py
+6 -3
View File
@@ -183,7 +183,8 @@ class FaceStylizer(base_vision_task_api.BaseVisionTaskApi):
ValueError: If any of the input arguments is invalid.
RuntimeError: If face stylization failed to run.
"""
normalized_rect = self.convert_to_normalized_rect(image_processing_options)
normalized_rect = self.convert_to_normalized_rect(
image_processing_options, image)
output_packets = self._process_image_data({
_IMAGE_IN_STREAM_NAME: packet_creator.create_image(image),
_NORM_RECT_STREAM_NAME: packet_creator.create_proto(
@@ -221,7 +222,8 @@ class FaceStylizer(base_vision_task_api.BaseVisionTaskApi):
ValueError: If any of the input arguments is invalid.
RuntimeError: If face stylization failed to run.
"""
normalized_rect = self.convert_to_normalized_rect(image_processing_options)
normalized_rect = self.convert_to_normalized_rect(
image_processing_options, image)
output_packets = self._process_video_data({
_IMAGE_IN_STREAM_NAME: packet_creator.create_image(image).at(
timestamp_ms * _MICRO_SECONDS_PER_MILLISECOND
@@ -268,7 +270,8 @@ class FaceStylizer(base_vision_task_api.BaseVisionTaskApi):
ValueError: If the current input timestamp is smaller than what the face
stylizer has already processed.
"""
normalized_rect = self.convert_to_normalized_rect(image_processing_options)
normalized_rect = self.convert_to_normalized_rect(
image_processing_options, image)
self._send_live_stream_data({
_IMAGE_IN_STREAM_NAME: packet_creator.create_image(image).at(
timestamp_ms * _MICRO_SECONDS_PER_MILLISECOND