o V h @sDddlmZddlmZddlmZddlmZGdddeZdS) )Path)Model)FastSAMPredictor)FastSAMValidatorcs<eZdZdZd fdd Zd fdd Zed d ZZS) FastSAMaD FastSAM model interface for segment anything tasks. This class extends the base Model class to provide specific functionality for the FastSAM (Fast Segment Anything Model) implementation, allowing for efficient and accurate image segmentation. Attributes: model (str): Path to the pre-trained FastSAM model file. task (str): The task type, set to "segment" for FastSAM models. Examples: >>> from ultralytics import FastSAM >>> model = FastSAM("last.pt") >>> results = model.predict("ultralytics/assets/bus.jpg") FastSAM-x.ptcs:t|dkrd}t|jdvsJdtj|dddS)zDInitialize the FastSAM model with the specified pre-trained weights.z FastSAM.ptr>.yml.yamlz/FastSAM models only support pre-trained models.segment)modeltaskN)strrsuffixsuper__init__)selfr  __class__T/var/www/vscode/kcb/lib/python3.10/site-packages/ultralytics/models/fastsam/model.pyrs zFastSAM.__init__FNc  s*t||||d}tj||fd|i|S)a Perform segmentation prediction on image or video source. Supports prompted segmentation with bounding boxes, points, labels, and texts. The method packages these prompts and passes them to the parent class predict method. Args: source (str | PIL.Image | numpy.ndarray): Input source for prediction, can be a file path, URL, PIL image, or numpy array. stream (bool): Whether to enable real-time streaming mode for video inputs. bboxes (list): Bounding box coordinates for prompted segmentation in format [[x1, y1, x2, y2], ...]. points (list): Point coordinates for prompted segmentation in format [[x, y], ...]. labels (list): Class labels for prompted segmentation. texts (list): Text prompts for segmentation guidance. **kwargs (Any): Additional keyword arguments passed to the predictor. Returns: (list): List of Results objects containing the prediction results. )bboxespointslabelstextsprompts)dictrpredict) rsourcestreamrrrrkwargsrrrrr#szFastSAM.predictcCsdttdiS)z[Returns a dictionary mapping segment task to corresponding predictor and validator classes.r ) predictor validator)rr)rrrrtask_map:szFastSAM.task_map)r)FNNNN) __name__ __module__ __qualname____doc__rrpropertyr# __classcell__rrrrr s rN) pathlibrultralytics.engine.modelrrrvalrrrrrrs