fix sam2 models
This commit is contained in:
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user