o 0j@sxddlmZmZmZmZmZddlZddlm Z ddl m Z ddl m Z mZdd lmZdd lmZGd d d e ZdS) )AnyListOptionalSequenceTupleN)logging)DetRunnerPredictor) ReadImageToBatch)InstanceSegPostProcessInstanceSegResultcseZdZdZdddeeffddZdefdd Zde fd d Z d d Z dde e deefddZdee de efddZddZZS)InstanceSegRunnerPredictorzAInstanceSegRunnerPredictor that inherits from DetRunnerPredictor.N) thresholdrcs`tj|i|hd|_|j|jvr+tddt|jd|jddd||_ dS) aInitializes InstanceSegPredictor. Args: *args: Arbitrary positional arguments passed to the superclass. threshold (Optional[float], optional): The threshold for filtering out low-confidence predictions. Defaults to None, in which case will use default from the config file. **kwargs: Arbitrary keyword arguments passed to the superclass. >zPP-YOLOE_seg-Sz'Cascade-MaskRCNN-ResNet50-vd-SSLDv2-FPNZSOLOv2zCascade-MaskRCNN-ResNet50-FPNzInstance Segmentation Models: "z, a" only supports prediction with a batch_size of one, if you set the predictor with a batch_size larger than one, no error will occur, however, it will actually inference with a batch_size of one, which will lead to a slower inference speed. You are now using Global model_name.N) super__init__*model_names_only_supports_batchsize_of_onerrwarningjoinlistconfigr)selfrargskwargs __class__y/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddlex/inference/models/instance_segmentation/predictor.pyrs    z#InstanceSegRunnerPredictor.__init__returncCstS)zvReturns the result class, InstanceSegResult. Returns: type: The InstanceSegResult class. rrr"r"r#_get_result_class6sz,InstanceSegRunnerPredictor._get_result_classcCstddg}|jdD]'}|d}|j|}|d|}|r'||fi|n||}|r2||q |||}||fS)zBuild the preprocessors and postprocessors based on the configuration. Returns: tuple: A tuple containing the preprocessors and postprocessors. RGB)formatZ Preprocesstype)r rZ _FUNC_MAPpopappendbuild_to_batchbuild_postprocess)rpre_opscfgZtf_keyfuncroppost_opr"r"r#_build>s    z!InstanceSegRunnerPredictor._buildcCsd}t|dS)N)Zimg_sizeZimgZ scale_factors)ordered_required_keys)r )rr4r"r"r#r,Us z)InstanceSegRunnerPredictor.build_to_batch batch_datac s|j}|jddD]}||}q |jd|}|j|jvr@g}t|djdD]fdd|D}||}||q)n||}||} | | ||durT|n|j } |j |j dd|Ddd| Ddd| Dd 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, box and mask for every instance of the batch. Keys include 'input_path', 'input_img', 'boxes' and 'masks'. Nrcsg|]}|dqS))N.r").0Z batch_input_ir"r# vs z6InstanceSegRunnerPredictor.process..cSg|]}|dqS)Zori_imgr")r7datar"r"r#r:cSr;)boxesr"r7resultr"r"r#r:r=cSr;)masksr"r?r"r"r#r:r=)Z input_pathZ page_indexZ input_imgr>rA) Z instancesr.rrrangeshaperunnerr+_format_outputr2rZ input_pathsZ page_indexes) rr5rZdatasZpre_opZ batch_inputsZ batch_predsZ batch_inputs_Z batch_pred_Z preds_listZ boxes_masksr"r8r#process_s0           z"InstanceSegRunnerPredictor.processpredcsd}gtdtr/tddkr/ddDddDfddttDStdtrKtddkrKfddttDSgttd D](}d |}||}d||}|d ||}||}qUfd dttDS) a Transform batch outputs into a list of single image output. Args: pred (Sequence[Any]): The input predictions, which can be either a list of 3 or 4 elements. - When len(pred) == 4, it is expected to be in the format [boxes, class_ids, scores, masks], compatible with SOLOv2 output. - When len(pred) == 3, it is expected to be in the format [boxes, box_nums, masks], compatible with Instance Segmentation output. Returns: List[dict]: A list of dictionaries, each containing either 'class_id' and 'masks' (for SOLOv2), or 'boxes' and 'masks' (for Instance Segmentation), or just 'boxes' if no masks are provided. rrcSsg|] }|d|dgqS)r r r"r7Zpred_r"r"r#r:sz=InstanceSegRunnerPredictor._format_output..cSr;)r"rHr"r"r#r:r=c*g|]}t|t|dqS))Zclass_idrAnparrayr7r9) pred_class_id pred_maskr"r#r:s  rIcs2g|]}t|dt|ddqS)rr r>rArKrN)rGr"r#r:s$r r crJ)rQrKrN)pred_boxrPr"r#r:s) isinstancerlenrBr+)rrGZ box_idx_startidxZ np_boxes_numZ box_idx_endZnp_boxesZnp_masksr")rGrRrOrPr#rEs0         z)InstanceSegRunnerPredictor._format_outputcCst|jd|jddS)NZdraw_thresholdZ label_list)rlabels)rrr%r"r"r#r-sz,InstanceSegRunnerPredictor.build_postprocess)N)__name__ __module__ __qualname____doc__rfloatrr)r&rr3r,rrrFrdictrEr- __classcell__r"r"r r#rs /3r)typingrrrrrnumpyrLutilsrZobject_detectionr Zobject_detection.processorsr r Z processorsrr@rrr"r"r"r#s