WpfVdZddlZddlmZddlmZddlmZddlmZ ddl m Z ddl m Zdd l mZdd lmZdd lmZdd lmZdd lmZejZe jZejZejZ ej!Z"dZ#dZ$dZ%dZ&dZ'dZ(dZ)dZ*ej+GddZ,Gddej-Z.dS)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.FaceStylizerGraphcDeZdZUdZeed<ejdefdZ dS)FaceAlignerOptionsznOptions for the face aligner task. Attributes: base_options: Base options for the face aligner task. r returncb|j}d|_t|S)z0Generates a FaceStylizerOptions protobuf object.Fr)r to_pb2use_stream_mode_FaceStylizerGraphOptionsProto)selfbase_options_protos j/var/www/html/nettyfy-visnx/env/lib/python3.11/site-packages/mediapipe/tasks/python/vision/face_aligner.pyrzFaceAlignerOptions.to_pb2:s4*1133).& )7I J J JJN) __name__ __module__ __qualname____doc__ _BaseOptions__annotations__r do_not_generate_docsrrr rrr0s_ $K4KKK%$KKKr rceZdZdZededdfdZededdfdZ d de j d e e de j fd Z dS) FaceAlignerz-Class that performs face alignment on images. model_pathrclt|}t|}||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_pathEs8& << > HH/1KL M M HHj"89 : :    I 3''U'CC   r Nrr c `|||}|ttj|t tj|i}|t rdStj |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--~:5AA ;  " " $ $! ! /N 01::<< T  ">2L#M N NNr )N)r!r"r#r$ classmethodstrr1rr. image_moduleImager_ImageProcessingOptionsrNr(r rr*r*Bs55,c,m,,,;,,(:};FEI!O!O  !O!))@ A!O !O!O!O!O!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#"++++++******FFFFFFYYYYYYKKKKKKEEEEEEJJJJJJCCCCCCjjjjjj^^^^^^". #<#8 9P  % -,'"$  K KKKKKKKK"\O\O\O\O\O&8\O\O\O\O\Or