fix sam2 models

This commit is contained in:
2025-07-26 07:57:01 -07:00
parent aa6ee4d40e
commit 1bec8113de
5 changed files with 25 additions and 1 deletions

View File

@@ -27,6 +27,8 @@ class MattingConfig:
use_disparity_mapping: bool = True
memory_offload: bool = True
fp16: bool = True
sam2_model_cfg: str = "sam2_hiera_l.yaml"
sam2_checkpoint: str = "sam2_hiera_large.pt"
@dataclass
@@ -89,7 +91,9 @@ class VR180Config:
'matting': {
'use_disparity_mapping': self.matting.use_disparity_mapping,
'memory_offload': self.matting.memory_offload,
'fp16': self.matting.fp16
'fp16': self.matting.fp16,
'sam2_model_cfg': self.matting.sam2_model_cfg,
'sam2_checkpoint': self.matting.sam2_checkpoint
},
'output': {
'path': self.output.path,