|h >ddlmZddlmZddlmZGddeZy)) BasePredictor)Results)opsc(eZdZdZdZdZdZdZy)DetectionPredictora: A class extending the BasePredictor class for prediction based on a detection model. This predictor specializes in object detection tasks, processing model outputs into meaningful detection results with bounding boxes and class predictions. Attributes: args (namespace): Configuration arguments for the predictor. model (nn.Module): The detection model used for inference. batch (list): Batch of images and metadata for processing. Methods: postprocess: Process raw model predictions into detection results. construct_results: Build Results objects from processed predictions. construct_result: Create a single Result object from a prediction. get_obj_feats: Extract object features from the feature maps. Examples: >>> from ultralytics.utils import ASSETS >>> from ultralytics.models.yolo.detect import DetectionPredictor >>> args = dict(model="yolo11n.pt", source=ASSETS) >>> predictor = DetectionPredictor(overrides=args) >>> predictor.predict_cli() c t|dddu}tj||jj|jj |jj |jj|jj|jjdk(rdnt|jjt|jdd|jjdk(| }t|tstj|}|r$|j!|j"|d }|d}|j$|||fi|}|rt'|D] \}} | |_|S) ay Post-process predictions and return a list of Results objects. This method applies non-maximum suppression to raw model predictions and prepares them for visualization and further analysis. Args: preds (torch.Tensor): Raw predictions from the model. img (torch.Tensor): Processed input image tensor in model input format. orig_imgs (torch.Tensor | list): Original input images before preprocessing. **kwargs (Any): Additional keyword arguments. Returns: (list): List of Results objects containing the post-processed predictions. Examples: >>> predictor = DetectionPredictor(overrides=dict(model="yolo11n.pt")) >>> results = predictor.predict("path/to/image.jpg") >>> processed_results = predictor.postprocess(preds, img, orig_imgs) _featsNdetectrend2endFobb)max_detncr rotated return_idxs)getattrrnon_max_suppressionargsconfiouclasses agnostic_nmsr tasklenmodelnames isinstancelistconvert_torch2numpy_batch get_obj_featsr construct_resultszipfeats) selfpredsimg orig_imgskwargs save_feats obj_featsresultsrfs e/var/www/html/test/engine/venv/lib/python3.12/site-packages/ultralytics/models/yolo/detect/predict.py postprocesszDetectionPredictor.postprocess"s5*T8T2$> ''  IINN IIMM II   II " "II%%IINNh.qC 8H8H4IDJJ 59IINNe+"  )T*55i@I **4;;aAI!HE($((YI&I GY/ 1 c ddl}t|Dcgc]}|jdc}}|j|Dcgc]U}|j ddddj |jdd||jd|zj dWc}d}t||Dcgc]\}}t|r||ngc}}Scc}wcc}wcc}}w)z.Extract object features from the feature maps.rNr)dim) torchminshapecatpermutereshapemeanr"r) r$ feat_mapsidxsr6xsr*r#idxs r.r z DetectionPredictor.get_obj_featsTs Y// 0IIen o`aQYYq!Q " * *1771:r1aggajAo N S SXZ S [ ouv BEYPTAUV:5#c#hc B.VV 0 oVsC AC,Cc t|||jdDcgc]\}}}|j||||c}}}Scc}}}w)a Construct a list of Results objects from model predictions. Args: preds (List[torch.Tensor]): List of predicted bounding boxes and scores for each image. img (torch.Tensor): Batch of preprocessed images used for inference. orig_imgs (List[np.ndarray]): List of original images before preprocessing. Returns: (List[Results]): List of Results objects containing detection information for each image. r)r"batchconstruct_result)r$r%r&r'predorig_imgimg_paths r.r!z$DetectionPredictor.construct_results^sQ-0y$**Q-,P  (h  ! !$Xx @   sAc tj|jdd|ddddf|j|ddddf<t|||jj |ddddfS)a. Construct a single Results object from one image prediction. Args: pred (torch.Tensor): Predicted boxes and scores with shape (N, 6) where N is the number of detections. img (torch.Tensor): Preprocessed image tensor used for inference. orig_img (np.ndarray): Original image before preprocessing. img_path (str): Path to the original image file. Returns: (Results): Results object containing the original image, image path, class names, and scaled bounding boxes. r2N)pathrboxes)r scale_boxesr8rrr)r$rEr&rFrGs r.rDz#DetectionPredictor.construct_resultoshoociimT!RaR%[(..QQU xhdjj6F6FdSTVXWXVXSXkZZr0N)__name__ __module__ __qualname____doc__r/r r!rDr0r.rrs20dW "[r0rN)ultralytics.engine.predictorrultralytics.engine.resultsrultralytics.utilsrrrRr0r.rVs 7.!u[u[r0