13 lines
1.6 KiB
Markdown
13 lines
1.6 KiB
Markdown
|
| Argument | Type | Default | Description |
|
||
|
|
| ------------ | ----------- | -------------- | ------------------------------------------------------------------------------------ |
|
||
|
|
| `data` | `str` | required | Path to directory containing target images for annotation or segmentation. |
|
||
|
|
| `det_model` | `str` | `'yolo11x.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 mobile_sam 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). |
|