a yf @s<ddlmZddlmZddlmZmZGdddeZdS))Results)DetectionPredictor) DEFAULT_CFGopscs0eZdZdZeddffdd ZddZZS)SegmentationPredictora A class extending the DetectionPredictor class for prediction based on a segmentation model. Example: ```python from ultralytics.utils import ASSETS from ultralytics.models.yolo.segment import SegmentationPredictor args = dict(model="yolov8n-seg.pt", source=ASSETS) predictor = SegmentationPredictor(overrides=args) predictor.predict_cli() ``` Ncst|||d|j_dS)z`Initializes the SegmentationPredictor with the provided configuration, overrides, and callbacks.segmentN)super__init__argstask)selfcfgZ overrides _callbacks __class__c/var/www/html/django/DPS/env/lib/python3.9/site-packages/ultralytics/models/yolo/segment/predict.pyr szSegmentationPredictor.__init__c Cstj|d|jj|jj|jj|jjt|jj |jj d}t |t sLt |}g}t |dtrj|ddn|d}tt|||jdD]J\}\}} } t|sd} n|jjr,t|jdd|ddddf| j|ddddf<t|||ddddf|ddddf| jdd} n|tj|||ddddf|ddddf|jddd d } t|jdd|ddddf| j|ddddf<|t| | |jj |ddddf| d q|S) zVApplies non-max suppression and processes detections for each image in an input batch.r)Zagnosticmax_detncclassesNT)Zupsample)pathnamesZboxesmasks)rZnon_max_suppressionr confZiouZ agnostic_nmsrlenmodelrr isinstancelistZconvert_torch2numpy_batchtuple enumeratezipbatchZ retina_masksZ scale_boxesshapeZprocess_mask_nativeZ process_maskappendr) r predsZimgZ orig_imgspresultsprotoipredZorig_imgZimg_pathrrrr postprocesss.   "& :@B:.z!SegmentationPredictor.postprocess)__name__ __module__ __qualname____doc__rr r/ __classcell__rrrrrsrN)Zultralytics.engine.resultsrZ&ultralytics.models.yolo.detect.predictrZultralytics.utilsrrrrrrrs