30 lines
946 B
YAML
30 lines
946 B
YAML
input:
|
|
video_path: "/workspace/data/input_video.mp4"
|
|
|
|
processing:
|
|
scale_factor: 0.5 # A40 can handle 0.5 well
|
|
chunk_size: 600 # Category A.4: Larger chunks for better VRAM utilization (was 200)
|
|
overlap_frames: 30 # Reduced overlap
|
|
|
|
detection:
|
|
confidence_threshold: 0.7
|
|
model: "yolov8m" # Use medium model on A40
|
|
|
|
matting:
|
|
use_disparity_mapping: true
|
|
memory_offload: false # A40 has enough VRAM
|
|
fp16: true
|
|
sam2_model_cfg: "configs/sam2.1/sam2.1_hiera_l.yaml"
|
|
sam2_checkpoint: "segment-anything-2/checkpoints/sam2.1_hiera_large.pt"
|
|
|
|
output:
|
|
path: "/workspace/output/matted_video.mp4"
|
|
format: "greenscreen" # Changed to greenscreen for easier testing
|
|
background_color: [0, 255, 0]
|
|
maintain_sbs: true
|
|
preserve_audio: true # Category A.1: Audio preservation
|
|
verify_sync: true # Category A.2: Frame count validation
|
|
|
|
hardware:
|
|
device: "cuda"
|
|
max_vram_gb: 45 # A40 has 48GB, leave headroom |