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

simplify2

Browse Source
This commit is contained in:
Scott Register
2025-07-27 09:55:52 -07:00
parent 66895a87a0
commit e38f63f539
1 changed files with 11 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
vr180_streaming/sam2_streaming_simple.py
View File

@@ -28,10 +28,20 @@ class SAM2StreamingProcessor:
self.device = torch.device(config.get('hardware', {}).get('device', 'cuda')) self.device = torch.device(config.get('hardware', {}).get('device', 'cuda'))
# SAM2 model configuration # SAM2 model configuration
model_cfg = config.get('matting', {}).get('sam2_model_cfg', 'sam2.1_hiera_l') model_cfg_name = config.get('matting', {}).get('sam2_model_cfg', 'sam2.1_hiera_l')
checkpoint = config.get('matting', {}).get('sam2_checkpoint', checkpoint = config.get('matting', {}).get('sam2_checkpoint',
'segment-anything-2/checkpoints/sam2.1_hiera_large.pt') 'segment-anything-2/checkpoints/sam2.1_hiera_large.pt')
# Map config name to full path
config_mapping = {
'sam2.1_hiera_t': 'segment-anything-2/sam2/configs/sam2.1/sam2.1_hiera_t.yaml',
'sam2.1_hiera_s': 'segment-anything-2/sam2/configs/sam2.1/sam2.1_hiera_s.yaml',
'sam2.1_hiera_b+': 'segment-anything-2/sam2/configs/sam2.1/sam2.1_hiera_b+.yaml',
'sam2.1_hiera_l': 'segment-anything-2/sam2/configs/sam2.1/sam2.1_hiera_l.yaml',
}
model_cfg = config_mapping.get(model_cfg_name, model_cfg_name)
# Build predictor (simple, clean approach) # Build predictor (simple, clean approach)
self.predictor = build_sam2_video_predictor( self.predictor = build_sam2_video_predictor(
model_cfg, model_cfg,