o 0jw@sddlmZmZmZmZmZddlZddlm Z ddl m Z ddl m Z dd lmZmZmZdd lmZd d lmZmZd d lmZGdddeZdS))AnyDictListTupleUnionN) FuncRegisterImageBatchSampler) ReadImage) NormalizeToBatch ToCHWImage)RunnerPredictor)ResizeSegPostProcess SegResultc seZdZdZiZeeZ ddeee edfde de ddffdd Z de fd d Zdefd d Zde fd dZ dde eeejfdeee edfde eeffddZed  dddZed    dddZZS) SegRunnerPredictorz6SegRunnerPredictor that inherits from RunnerPredictor.N target_sizeargskwargsreturncs,tj|i|||_|\|_|_dS)aInitializes SegPredictor. Args: target_size: Image size used for inference. *args: Arbitrary positional arguments passed to the superclass. **kwargs: Arbitrary keyword arguments passed to the superclass. N)super__init__r_build preprocessorspostprocessers)selfrrr __class__y/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/semantic_segmentation/predictor.pyr"s zSegRunnerPredictor.__init__cCstS)zBuilds and returns an ImageBatchSampler instance. Returns: ImageBatchSampler: An instance of ImageBatchSampler. r r!r$r$r%_build_batch_sampler3sz'SegRunnerPredictor._build_batch_samplercCstS)zfReturns the result class, SegResult. Returns: type: The SegResult class. rr&r$r$r%_get_result_class;sz$SegRunnerPredictor._get_result_classc Csdtddi}t|d<|jddD]"}|d}|j|}|}|r+||fi|n||\}}|||<qt|d<d |vrN|jd |d d \}}||d <|jd urc|jd ||jd \}}||d <t} || fS) zBuild the preprocessors and postprocessors based on the configuration. Returns: tuple: A tuple containing the preprocessors and postprocessors. ReadRGB)formatToCHWZDeployZ transformstyperr)rN)r rconfigpop _FUNC_MAPrrr) r!rcfgZtf_keyfuncrnameop_r r$r$r%rCs"       zSegRunnerPredictor._build batch_datacCs|jd|jd}|jd||d}|jd|d}|jd|d}|jd|d}|j|d}t|d krBtj|d t|d d }|||}|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). target_size: Image size used for inference. Returns: dict: A dictionary containing the input path, raw image, and predicted segmentation maps for every instance of the batch. Keys include 'input_path', 'input_img', and 'pred'. r))imgsr)r8rrr,r)xrr)Zaxis)Z input_pathZ page_indexZ input_imgpred) rZ instancesrunnerlennpsplitr Z input_pathsZ page_indexes)r!r7rZbatch_raw_imgsZ batch_imgsr9Z batch_predsr$r$r%process]s    zSegRunnerPredictor.processr?cCst||d}d|fS)N)meanstdr)r)r!rArBr5r$r$r%build_normalizes z"SegRunnerPredictor.build_normalizerr.T LINEARcCst||||d}d|fS)N)r keep_ratio size_divisorinterpr)r)r!rrFrGrHr5r$r$r% build_resizeszSegRunnerPredictor.build_resize)N)r@r@)r.TrDrE)__name__ __module__ __qualname____doc__r1rregisterrintrrrrr r'r-r(rstrr=Zndarrayrr?rCrI __classcell__r$r$r"r%rsF  $ r)typingrrrrrnumpyr=Zutils.func_registerrZcommon.batch_samplerr Z common.readerr commonrrrZ predictorsrZ processorsrrresultrrr$r$r$r%s