WpfLdZddlZddlmZddlmZddlmZddlmZ ddl m Z ddl m Zdd l mZdd lmZdd lmZdd lmZdd lmZejZe jZejZejZ dZ!dZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)ej*GddZ+Gddej,Z-dS)zMediaPipe face stylizer task.N)Optional)packet_creator) packet_getter)image)face_stylizer_graph_options_pb2 base_options) task_info) doc_controls)base_vision_task_api)image_processing_options)vision_task_running_modestylized_imageSTYLIZED_IMAGE norm_rect_in NORM_RECTimage_in image_outIMAGEz6mediapipe.tasks.vision.face_stylizer.FaceStylizerGraphicDeZdZUdZeed<ejdefdZ dS)FaceStylizerOptionszpOptions for the face stylizer task. Attributes: base_options: Base options for the face stylizer task. r returncT|j}t|S)z1Generates an FaceStylizerOptions protobuf object.r)r to_pb2_FaceStylizerGraphOptionsProto)selfbase_options_protos k/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/mediapipe/tasks/python/vision/face_stylizer.pyrzFaceStylizerOptions.to_pb2:s+*1133 )7I J J JJN) __name__ __module__ __qualname____doc__ _BaseOptions__annotations__r do_not_generate_docsrrrrrr0s_ $K4KKK%$KKKrrceZdZdZededdfdZededdfdZ d de j d e e de j fd Z dS) FaceStylizerz/Class that performs face stylization on images. model_pathrclt|}t|}||S)aECreates an `FaceStylizer` object from a TensorFlow Lite model and the default `FaceStylizerOptions`. Note that the created `FaceStylizer` instance is in image mode, for stylizing one face on a single image input. Args: model_path: Path to the model. Returns: `FaceStylizer` object that's created from the model file and the default `FaceStylizerOptions`. Raises: ValueError: If failed to create `FaceStylizer` object from the provided file such as invalid file path. RuntimeError: If other types of error occurred. )model_asset_pathr)r$rcreate_from_options)clsr*r optionss rcreate_from_model_pathz#FaceStylizer.create_from_model_pathDs8& << > HH)+?@ A A HHj"89 : :    I 3''))(>D   rNrr c `|||}|ttj|t tj|i}|t rdStj |tS)aDPerforms face stylization on the provided MediaPipe Image. Only use this method when the FaceStylizer is created with the image running mode. Args: image: MediaPipe Image. image_processing_options: Options for image processing. Returns: The stylized image of the most visible face. The stylized output image size is the same as the model output size. None if no face is detected on the input image. Raises: ValueError: If any of the input arguments is invalid. RuntimeError: If face stylization failed to run. N) convert_to_normalized_rect_process_image_datar<r create_imager> create_protorr@is_emptyr get_image)rrr normalized_rectoutput_packetss rstylizezFaceStylizer.stylize|s.55 %O--~:5AA ;  " " $ $! ! /N *+4466 T  ">2F#G H HHr)N)r r!r"r# classmethodstrr0rr- image_moduleImager_ImageProcessingOptionsrNr'rrr)r)As77,c,n,,,;,,(;;FEI"I"I  "I!))@ A"I "I"I"I"I"I"Irr)).r# dataclassestypingrmediapipe.pythonrr$mediapipe.python._framework_bindingsrrQ-mediapipe.tasks.cc.vision.face_stylizer.protormediapipe.tasks.python.corer base_options_moduler task_info_module1mediapipe.tasks.python.core.optional_dependenciesr "mediapipe.tasks.python.vision.corer r image_processing_options_modulerrC BaseOptionsr$FaceStylizerGraphOptionsrImageProcessingOptionsrSTaskInfor8r@r?r>r=r<rAr;r9_MICRO_SECONDS_PER_MILLISECOND dataclassrBaseVisionTaskApir)r'rrrfs$#++++++******FFFFFFYYYYYYKKKKKKEEEEEEJJJJJJCCCCCCjjjjjj^^^^^^". #<:P  % '&'"$  K!%  K K K K K K K K ]I]I]I]I]I'9]I]I]I]I]Ir