fix sam2 models

2025-07-26 07:57:01 -07:00
parent aa6ee4d40e
commit 1bec8113de
5 changed files with 25 additions and 1 deletions
--- a/vr180_matting/video_processor.py
+++ b/vr180_matting/video_processor.py
@@ -51,6 +51,8 @@ class VideoProcessor:
            
            # Initialize SAM2 model
            self.sam2_model = SAM2VideoMatting(
+                model_cfg=self.config.matting.sam2_model_cfg,
+                checkpoint_path=self.config.matting.sam2_checkpoint,
                device=self.config.hardware.device,
                memory_offload=self.config.matting.memory_offload,
                fp16=self.config.matting.fp16