scott/test2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

more stuff

Browse Source
This commit is contained in:
Scott Register
2025-07-27 08:19:42 -07:00
parent 7431954482
commit 9faaf4ed57
5 changed files with 21 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
vr180_streaming/sam2_streaming.py
View File

@@ -34,6 +34,11 @@ class SAM2StreamingProcessor:
self.config = config
self.device = torch.device(config.get('hardware', {}).get('device', 'cuda'))
# Processing parameters (set before _init_predictor)
self.memory_offload = config.get('matting', {}).get('memory_offload', True)
self.fp16 = config.get('matting', {}).get('fp16', True)
self.correction_interval = config.get('matting', {}).get('correction_interval', 300)
# SAM2 model configuration
model_cfg = config.get('matting', {}).get('sam2_model_cfg', 'sam2.1_hiera_l')
checkpoint = config.get('matting', {}).get('sam2_checkpoint',
@@ -43,11 +48,6 @@ class SAM2StreamingProcessor:
self.predictor = None
self._init_predictor(model_cfg, checkpoint)
# Processing parameters
self.memory_offload = config.get('matting', {}).get('memory_offload', True)
self.fp16 = config.get('matting', {}).get('fp16', True)
self.correction_interval = config.get('matting', {}).get('correction_interval', 300)
# State management
self.states = {} # eye -> inference state
self.object_ids = []