o 0j @s@ddlmZmZmZmZddlZddlmZGdddeZ dS))AnyDictListUnionN)ImageFeatureRunnerPredictorc s`eZdZdZdddedededdffd d Zd eee e j fdee e ffd d Z ZS)FaceFeaturePredictorzDFaceFeaturePredictor that inherits from ImageFeatureRunnerPredictor.F)flipargsr kwargsreturnNcstj|i|||_dS)a!Initializes ClasPredictor. Args: *args: Arbitrary positional arguments passed to the superclass. flip: Whether to perform face flipping during inference. Default is False. **kwargs: Arbitrary keyword arguments passed to the superclass. N)super__init__r )selfr r r  __class__p/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/face_feature/predictor.pyrs zFaceFeaturePredictor.__init__ batch_datac Cs|jd|jd}|jd|d}|jd|d}|jd|d}|jd|d}|j|d}|jrP|jdd |Dd}tt|D] }||||||<qC|jd |}|j|j||d S) a Process a batch of data through the preprocessing, inference, and postprocessing. Args: batch_data (List[Union[str, np.ndarray], ...]): A batch of input data (e.g., image file paths). Returns: dict: A dictionary containing the input path, raw image, class IDs, scores, and label names for every instance of the batch. Keys include 'input_path', 'input_img', 'class_ids', 'scores', and 'label_names'. ZRead)ZimgsZResizeZ NormalizeZToCHWZToBatch)xcSsg|] }tj|ddqS))Zaxis)npr ).0datarrr 5sz0FaceFeaturePredictor.process..ZNormalizeFeatures)Z input_pathZ page_indexZ input_imgfeature) Z preprocessorsZ instancesrunnerr rangelenZpostprocessorsZ input_pathsZ page_indexes) rrZbatch_raw_imgsZ batch_imgsrZ batch_predsZbatch_preds_flippedifeaturesrrrprocess$s   zFaceFeaturePredictor.process)__name__ __module__ __qualname____doc__rboolrrrstrrZndarrayrr! __classcell__rrrrrs$0 r) typingrrrrnumpyrZ image_featurerrrrrrs