52 lines
2.2 KiB
YAML
52 lines
2.2 KiB
YAML
# YOLO Annotator Configuration - CPU Only (ONNX)
|
|
#
|
|
# This configuration uses ONNX Runtime for CPU-only inference.
|
|
# No GPU required - works on any system.
|
|
#
|
|
# Usage:
|
|
# python scripts/annotate.py --config configs/annotator_cpu.yaml
|
|
|
|
model:
|
|
#path: "models/krg_tuang_yolov9t_best.pt" # ONNX model file
|
|
#path: "models/tuangatas.pt" # ONNX model file
|
|
path: "models/krg_masuk_yolov9t_best.pt" # ONNX model file
|
|
device: "cpu" # cpu (ONNX uses CPU by default)
|
|
backend: "pt" # Force ONNX backend
|
|
conf_threshold: 0.25 # Confidence threshold
|
|
iou_threshold: 0.45 # NMS IoU threshold
|
|
|
|
# ONNX specific options
|
|
onnx:
|
|
num_threads: 0 # CPU threads (0 = auto)
|
|
optimization_level: "all" # Graph optimization level
|
|
|
|
video:
|
|
source: "input/karung_masuk.mp4" # Video file path
|
|
#source: "input/tuang_train_20260120-1.mp4" # Video file path
|
|
#source: "input/atas.mp4" # Video file path
|
|
sample_fps: 3 # Frames per second to extract
|
|
max_frames: null # Max frames (null = all)
|
|
start_time: 0 # Start time in seconds
|
|
end_time: null # End time (null = end of video)
|
|
resize: null # [width, height] or null
|
|
|
|
detection:
|
|
classes: 0 # Class IDs to keep (null = all)
|
|
min_confidence: 0.3 # Minimum confidence to save
|
|
min_area: 100 # Minimum bbox area in pixels
|
|
max_area: null # Maximum bbox area (null = no limit)
|
|
min_size: 0.01 # Minimum bbox dimension (normalized)
|
|
|
|
output:
|
|
directory: "output/annotations" # Output directory
|
|
save_snapshots: true # Save clean images
|
|
save_labels: true # Save YOLO labels
|
|
save_debug: true # Save debug visualizations
|
|
save_manifest: true # Save JSON manifest
|
|
image_format: "jpg" # jpg or png
|
|
image_quality: 95 # JPEG quality (1-100)
|
|
|
|
# Class names (COCO subset - common objects)
|
|
class_names:
|
|
0: karung
|