|h ddlmZddlmZmZmZddlmZmZ ddee efde de de d e d e d e d e d eee deee efddfdZ y))Path)ListOptionalUnion)SAMYOLONdata det_model sam_modeldeviceconfiouimgszmax_detclasses output_dirreturnc ~t|}t|}t|}| s|j|jdz } t| j dd||d||||||} | D]?} | j jjj} | 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)a 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)streamr r rrrrF)bboxesverbosesaver rz.txtwzutf-8)encoding  N)rrrparentstemmkdirboxesclsinttolistxyxyorig_imgmasksxynopenpath enumerateanymapstrreshapewritejoin)r r r r r rrrrr det_resultsresult class_idsr$ sam_resultssegmentsfissegments Y/var/www/html/test/engine/venv/lib/python3.12/site-packages/ultralytics/data/annotator.py auto_annotater? sFYIII :D [[dii[0E#FF D$7 T&tESZdkK OLL$$((*113 LL%%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 yolo11x.ptzsam_b.ptg?g?ii,NN) pathlibrtypingrrr ultralyticsrrr1floatr&r?r>rGs((! "#'-1:O T :O:O:O  :O  :O  :O :O:Od3i :OsDy)*:O :OrF