add sam2 yolo auto annotation

2026-02-04 15:29:36 +07:00
parent 7e56948ece
commit 5a951d8812
2061 changed files with 316473 additions and 0 deletions
@@ -0,0 +1,105 @@
+# Frigate-Mini Configuration - CPU Only (ONNX)
+# 
+# This configuration uses ONNX Runtime for CPU-only inference.
+# No special hardware required - works on any x86 or ARM system.
+#
+# Usage:
+#   python scripts/frigate_mini.py --config configs/frigate_mini_cpu.yaml
+
+# Global settings
+debug: true                           # Enable debug mode
+log_level: "info"                     # debug, info, warning, error
+
+# Detector configuration - ONNX CPU Mode
+detector:
+  type: "pt"                        # Use PYTORCH Runtime (CPU)
+  model_path: "models/krg_masuk_yolov9t_best.pt"   # PT model file
+  input_size: [640, 640]              # Model input resolution [width, height]
+  conf_threshold: 0.25                # Detection confidence threshold
+  nms_threshold: 0.45                 # NMS IoU threshold
+  
+  # ONNX specific settings
+  onnx:
+    device: "cpu"                     # cpu or cuda
+    num_threads: 0                    # CPU threads (0 = auto, uses all cores)
+    optimization_level: "all"         # none, basic, extended, all
+  
+  # No fallback needed for ONNX (it's already the most compatible)
+  fallback:
+    enabled: false
+
+# Video sources (cameras)
+cameras:
+  default:
+    enabled: true
+    source: "input/video.mp4"         # MP4 file path
+    fps: 5                            # Processing FPS (adjust based on CPU speed)
+    loop: true                        # Loop video playback
+    
+    detect:
+      enabled: true
+      width: 1280                     # Processing resolution
+      height: 720
+      
+    objects:
+      track:
+        - person
+        - car
+        - dog
+        - cat
+      filters:
+        person:
+          min_area: 1000
+          min_score: 0.4
+        car:
+          min_area: 2000
+          min_score: 0.35
+
+# Snapshot settings
+snapshots:
+  enabled: true
+  output_dir: "output/snapshots"
+  
+  trigger:
+    objects:
+      - person
+      - car
+    min_score: 0.5
+    cooldown: 2.0
+    
+  format: "jpg"
+  quality: 95
+  clean: true                         # No annotations on snapshot
+
+# Annotation export
+annotations:
+  enabled: true
+  output_dir: "output/labels"
+  format: "yolo"
+  pair_with_snapshots: true
+  min_score: 0.3
+
+# Debug output
+debug_output:
+  enabled: true
+  output_dir: "output/debug"
+  
+  object_list:
+    enabled: true
+    show_confidence: true
+    show_class: true
+    show_bbox: true
+    
+  visualization:
+    enabled: true
+    draw_boxes: true
+    draw_labels: true
+    save_interval: 10                 # Save every 10 frames
+    
+  stats:
+    show_fps: true
+    log_interval: 100
+
+# Class names (COCO)
+class_names:
+  0: karung