ukidZddlZddlmZmZddlZddlmZddlm Z ddl m Z ddl m Z dZd Zd ej d d d ede j"j$ffdZGdd ej(ZGdde Zy)zMediaPipe Face Detection.N) NamedTupleUnion) detection_pb2)location_data_pb2)face_detection_pb2) SolutionBasezHmediapipe/modules/face_detection/face_detection_short_range_cpu.binarypbzGmediapipe/modules/face_detection/face_detection_full_range_cpu.binarypb detectionkey_point_enum FaceKeyPointreturncR|r |jsy|jj|S)zA convenience method to return a face key point by the FaceKeyPoint type. Args: detection: A detection proto message that contains face key points. key_point_enum: A FaceKeyPoint type. Returns: A RelativeKeypoint proto message. N) location_datarelative_keypoints)r r s d/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/mediapipe/python/solutions/face_detection.py get_key_pointrs) )11   3 3N CCc(eZdZdZdZdZdZdZdZdZ y) r z3The enum type of the six face detection key points.rN) __name__ __module__ __qualname____doc__ RIGHT_EYELEFT_EYENOSE_TIP MOUTH_CENTERRIGHT_EAR_TRAGIONLEFT_EAR_TRAGIONrrr r .s%;) ( (,rcLeZdZdZdfd Zdej deffd ZxZ S) FaceDetectionzMediaPipe Face Detection. MediaPipe Face Detection processes an RGB image and returns a list of the detected face location data. Please refer to https://solutions.mediapipe.dev/face_detection#python-solution-api for usage examples. c|dk(rtnt}t| ||j t j d|idgy)a:Initializes a MediaPipe Face Detection object. Args: min_detection_confidence: Minimum confidence value ([0.0, 1.0]) for face detection to be considered successful. See details in https://solutions.mediapipe.dev/face_detection#min_detection_confidence. model_selection: 0 or 1. 0 to select a short-range model that works best for faces within 2 meters from the camera, and 1 for a full-range model best for faces within 5 meters. See details in https://solutions.mediapipe.dev/face_detection#model_selection. rmin_score_thresh detections)binary_graph_path graph_optionsoutputsN)_FULL_RANGE_GRAPH_FILE_PATH_SHORT_RANGE_GRAPH_FILE_PATHsuper__init__create_graph_optionsrFaceDetectionOptions)selfmin_detection_confidencemodel_selectionr) __class__s rr/zFaceDetection.__init__CsX8G!7K3Qm G+//  3 3 5"$<8  rimager c(t|d|iS)aProcesses an RGB image and returns a list of the detected face location data. Args: image: An RGB image represented as a numpy ndarray. Raises: RuntimeError: If the underlying graph throws any error. ValueError: If the input image is not three channel RGB. Returns: A NamedTuple object with a "detections" field that contains a list of the detected face location data. r6) input_data)r.process)r2r6r5s rr9zFaceDetection.processZs 7?w&6? 77r)g?r) rrrrr/npndarrayrr9 __classcell__)r5s@rr%r%8s) .82::8*88rr%)renumtypingrrnumpyr:mediapipe.framework.formatsrr mediapipe.modules.face_detectionrmediapipe.python.solution_baserr-r, Detection LocationDataRelativeKeypointrIntEnumr r%r#rrrGs $59?7igD&&D8FD 4"//@@ @AD"4<<18L18r