28 lines
729 B
YAML
28 lines
729 B
YAML
input:
|
|
video_path: "/workspace/data/input_video.mp4"
|
|
|
|
processing:
|
|
scale_factor: 0.5 # A40 can handle 0.5 well
|
|
chunk_size: 0 # Auto-calculate based on A40's 48GB VRAM
|
|
overlap_frames: 60
|
|
|
|
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: "alpha"
|
|
background_color: [0, 255, 0]
|
|
maintain_sbs: true
|
|
|
|
hardware:
|
|
device: "cuda"
|
|
max_vram_gb: 45 # A40 has 48GB, leave headroom |