Fi rddlmZddlmZddlmZmZ d ddZy)) annotations)Path)SAMYOLONc ~t|}t|}t|}| s|j|jdz } t| j dd||d||||||} | D]?} | j jjjx} s9| j j} || j| dd|}|djj}tt| t| jjz dd d 5}t!|D]o\}}|j#st%t&|j)d j}|j+| |d d j-|zdzq dddBy#1swYNxYw)aAutomatically annotate images using a YOLO object detection model and a SAM segmentation model. This function processes images in a specified directory, detects objects using a YOLO model, and then generates segmentation masks using a SAM model. The resulting annotations are saved as text files in YOLO format. Args: data (str | Path): Path to a folder containing images to be annotated. det_model (str): Path or name of the pre-trained YOLO detection model. sam_model (str): Path or name of the pre-trained SAM segmentation model. device (str): Device to run the models on (e.g., 'cpu', 'cuda', '0'). Empty string for auto-selection. conf (float): Confidence threshold for detection model. iou (float): IoU threshold for filtering overlapping boxes in detection results. imgsz (int): Input image resize dimension. max_det (int): Maximum number of detections per image. classes (list[int], optional): Filter predictions to specified class IDs, returning only relevant detections. output_dir (str | Path, optional): Directory to save the annotated results. If None, creates a default directory based on the input data path. Examples: >>> from ultralytics.data.annotator import auto_annotate >>> auto_annotate(data="ultralytics/assets", det_model="yolo26n.pt", sam_model="mobile_sam.pt") _auto_annotate_labelsT)exist_okparents)streamdeviceconfiouimgszmax_detclassesF)bboxesverbosesaver rz.txtwzutf-8)encoding  N)rrrparentstemmkdirboxesclsinttolistxyxyorig_imgmasksxynopenpath enumerateanymapstrreshapewritejoin)data det_model sam_modelr r rrrr output_dir det_resultsresult class_idsr sam_resultssegmentsfissegments \/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/ultralytics/data/annotator.py auto_annotater< sDYIII :D [[dii[0E#FF D$7 T&tESZdkK O ((,,.557 79 7LL%%E#FOOE5W\eklK"1~++//Hj)D,=,B,BBC4H#X_` Ode%h/ODAquuw"%c199R=+?+?+A"B9Q<. 2SXXg5F F MNO O O O  O Os'"F2 AF22F< ) z yolo26x.ptzsam_b.ptg?g?ii,NN)r.z str | Pathr/r*r0r*r r*r floatrr>rrrrrzlist[int] | Noner1zstr | Path | NonereturnNone) __future__rpathlibr ultralyticsrrr<r;rFs#! " $$(8O 8O8O8O  8O  8O  8O 8O8O8O"8O 8OrE