j JddlmZddlmZddlmZmZ dd dZd S)!) annotations)Path)SAMYOLO yolo26x.ptsam_b.pt??,Ndata str | Path det_modelstr sam_modeldeviceconffloatiouimgszintmax_detclasseslist[int] | None output_dirstr | Path | NonereturnNonec t|}t|}t|}| s|j|jdz } t| dd||d||||||} | D]X} | jj x} r"| jj } || j | dd|}|dj j }tt| t| jjz dd d 5}t!|D]\}}|rqt%t&|d  }|| |d d |zdz dddn #1swxYwYZdS)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)streamrrrrrrF)bboxesverbosesaverrz.txtwzutf-8)encoding  N)rrrparentstemmkdirboxesclsrtolistxyxyorig_imgmasksxynopenpath enumerateanymaprreshapewritejoin)rrrrrrrrrr det_resultsresult class_idsr0 sam_resultssegmentsfissegments _/home/longshao/multi-rider-rag/.venv/lib/python3.11/site-packages/ultralytics/data/annotator.py auto_annotaterI s=DYIII ::D G[di#F#F#FF D$777) T&tESZdkK O O (,,..5577 79 OL%E#)FOE5W\eklllK"1~+/Hj))D,=,=,BBHHH#X_``` Ode%h//OODAquuwwO"%c199R==+?+?+A+A"B"B9Q< 2 2 2SXXg5F5F F MNNNO O O O O O O O O O O O O O O O O Os BGG G ) rrr r r r r NN)rrrrrrrrrrrrrrrrrrrrrr) __future__rpathlibr ultralyticsrrrIrHrOs#"""""!!!!!!!! " $$(8O8O8O8O8O8O8OrN