76 lines
2.2 KiB
YAML
76 lines
2.2 KiB
YAML
# YOLO + SAM2 Video Processing Configuration
|
|
|
|
input:
|
|
video_path: "/path/to/input/video.mp4"
|
|
|
|
output:
|
|
directory: "/path/to/output/"
|
|
filename: "processed_video.mp4"
|
|
|
|
processing:
|
|
# Duration of each video segment in seconds
|
|
segment_duration: 5
|
|
|
|
# Scale factor for SAM2 inference (0.5 = half resolution)
|
|
inference_scale: 0.5
|
|
|
|
# YOLO detection confidence threshold
|
|
yolo_confidence: 0.6
|
|
|
|
# Which segments to run YOLO detection on
|
|
# Options: "all", [0, 5, 10], or [] for default (all)
|
|
detect_segments: "all"
|
|
|
|
# VR180 separate eye processing mode (default: false for backward compatibility)
|
|
separate_eye_processing: false
|
|
|
|
# Enable full greenscreen fallback when no humans detected (only used with separate_eye_processing)
|
|
enable_greenscreen_fallback: true
|
|
|
|
# Pixel overlap between left/right eyes for blending (optional, default: 0)
|
|
eye_overlap_pixels: 0
|
|
|
|
models:
|
|
# YOLO detection mode: "detection" (bounding boxes) or "segmentation" (direct masks)
|
|
yolo_mode: "segmentation" # Default: existing behavior, Options: "detection", "segmentation"
|
|
|
|
# YOLO model paths for different modes
|
|
yolo_detection_model: "models/yolo/yolo11l.pt" # Regular YOLO for detection mode
|
|
yolo_segmentation_model: "models/yolo/yolo11x-seg.pt" # Segmentation YOLO for segmentation mode
|
|
|
|
# SAM2 model configuration
|
|
sam2_checkpoint: "models/sam2/checkpoints/sam2.1_hiera_small.pt"
|
|
sam2_config: "models/sam2/configs/sam2.1/sam2.1_hiera_s.yaml"
|
|
|
|
video:
|
|
# Use NVIDIA hardware encoding (requires NVENC-capable GPU)
|
|
use_nvenc: true
|
|
|
|
# Output video bitrate
|
|
output_bitrate: "50M"
|
|
|
|
# Preserve original audio track
|
|
preserve_audio: true
|
|
|
|
# Force keyframes for better segment boundaries
|
|
force_keyframes: true
|
|
|
|
advanced:
|
|
# Green screen color (RGB values)
|
|
green_color: [0, 255, 0]
|
|
|
|
# Blue screen color for second object (RGB values)
|
|
blue_color: [255, 0, 0]
|
|
|
|
# YOLO human class ID (0 for COCO person class)
|
|
human_class_id: 0
|
|
|
|
# GPU memory management
|
|
cleanup_intermediate_files: true
|
|
|
|
# Logging level (DEBUG, INFO, WARNING, ERROR)
|
|
log_level: "INFO"
|
|
|
|
# Save debug frames with YOLO detections visualized
|
|
save_yolo_debug_frames: true
|