ukidZddlZddlmZddlmZddlmZddlmZ ddl m Z ddl m Zdd l mZdd lmZdd lmZdd lmZdd lmZej2Ze j6Zej:Zej>Z ejBZ"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*ejVGddZ,GddejZZ.y)zMediaPipe face aligner 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_modeface_alignmentFACE_ALIGNMENT norm_rect_in NORM_RECTimage_in image_outIMAGEz6mediapipe.tasks.vision.face_stylizer.FaceStylizerGraphcFeZdZUdZeed<ejdefdZ y)FaceAlignerOptionsznOptions for the face aligner task. Attributes: base_options: Base options for the face aligner task. r returnc\|jj}d|_t|S)z0Generates a FaceStylizerOptions protobuf object.Fr)r to_pb2use_stream_mode_FaceStylizerGraphOptionsProto)selfbase_options_protos e/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/mediapipe/tasks/python/vision/face_aligner.pyrzFaceAlignerOptions.to_pb2:s.**113).& )7I JJN) __name__ __module__ __qualname____doc__ _BaseOptions__annotations__r do_not_generate_docsrrr rrr0s2 $$K4K%Kr rceZdZdZededdfdZededdfdZ d de jd e e de jfd Z y) FaceAlignerz-Class that performs face alignment on images. model_pathrcTt|}t|}|j|S)aYCreates a `FaceAligner` object from a face landmarker task bundle and the default `FaceAlignerOptions`. Note that the created `FaceAligner` instance is in image mode, for aligning one face on a single image input. Args: model_path: Path to the face landmarker task bundle. Returns: `FaceAligner` object that's created from the model file and the default `FaceAlignerOptions`. Raises: ValueError: If failed to create `FaceAligner` 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"FaceAligner.create_from_model_pathEs)&  HH/1KL M HHj"89 :  I ''U'C  r Nrr c 0|j||}|jttj|t tj |ji}|tjrytj|tS)a#Performs face alignment on the provided MediaPipe Image. Only use this method when the FaceAligner is created with the image running mode. Args: image: MediaPipe Image. image_processing_options: Options for image processing. Returns: The aligned face image. The aligned 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 alignment failed to run. N) convert_to_normalized_rect_process_image_datar=r create_imager? create_protorrAis_emptyr get_image)rrr normalized_rectoutput_packetss ralignzFaceAligner.align}s,55 %O--~::5A ; ;  " " $! /N 01::<   " ">2L#M NNr )N)r!r"r#r$ classmethodstrr1rr. image_moduleImager_ImageProcessingOptionsrNr(r rr*r*Bs5,c,m,,,(:}FEI!O   !O!))@ A!O !Or r*)/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_modulerrunning_mode_module BaseOptionsr%FaceStylizerGraphOptionsrVisionTaskRunningModerDImageProcessingOptionsrSTaskInfor9rAr@r?r>r=rBr<r: dataclassrBaseVisionTaskApir*r(r rrgs#+*FYKEJCj^".. #<<#88 9PP  % % -,'"$  K KKK"\O&88\Or