h vSSKJr SSKJr SSKJrJr SSSjjrg)) annotations)Path)SAMYOLONc [U5n[U5n[U5nU (dURURS3- n [U 5R SSS9 U"USX4XVXxS9n U GHGn U R RR5R5=n (dM?U R Rn U"U RU SSUS9nUSRRn[[U 5[U R5R- S3S S S 9n[!U5HvunnUR#5(dM[%[&UR)S 5R55nUR+U US 3S R-U5-S-5 Mx SSS5 GMJ g!,(df  GM]=f)aC Automatically 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="yolo11n.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 T/home/james-whalen/.local/lib/python3.13/site-packages/ultralytics/data/annotator.py auto_annotater< sFYIII :D [[dii[0E#FF D$7 T&SZK ((,,.557 79 7LL%%E#FOOE5W\eklK"1~++//Hj)D,=,B,BBC4H#X_`de%h/DAquuww"%c199R=+?+?+A"B9Q<. 2SXXg5F F MN0a`  a`s#%F7 AF77 G ) z yolo11x.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#! " $$(9O 9O9O9O  9O  9O  9O 9O9O9O"9O 9OrE