mirror of
https://github.com/ultralytics/ultralytics
synced 2026-05-12 18:14:20 +00:00
Signed-off-by: Glenn Jocher <glenn.jocher@ultralytics.com> Co-authored-by: Laughing-q <1185102784@qq.com> Co-authored-by: UltralyticsAssistant <web@ultralytics.com>
1.6 KiB
1.6 KiB
| Argument | Type | Default | Description |
|---|---|---|---|
data |
str |
required | Path to directory containing target images for annotation or segmentation. |
det_model |
str |
'yolo26x.pt' |
YOLO detection model path for initial object detection. |
sam_model |
str |
'sam_b.pt' |
SAM model path for segmentation (supports SAM, SAM2 variants, and MobileSAM models). |
device |
str |
'' |
Computation device (e.g., 'cuda:0', 'cpu', or '' for automatic device detection). |
conf |
float |
0.25 |
YOLO detection confidence threshold for filtering weak detections. |
iou |
float |
0.45 |
IoU threshold for Non-Maximum Suppression to filter overlapping boxes. |
imgsz |
int |
640 |
Input size for resizing images (must be multiple of 32). |
max_det |
int |
300 |
Maximum number of detections per image for memory efficiency. |
classes |
list[int] |
None |
List of class indices to detect (e.g., [0, 1] for person & bicycle). |
output_dir |
str |
None |
Save directory for annotations (defaults to './labels' relative to data path). |