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

bullshit

Browse Source
This commit is contained in:
Scott Register
2025-07-27 09:23:15 -07:00
parent 7b3ffb7830
commit 9b7f36fec2
1 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
vr180_streaming/sam2_streaming.py
View File

@@ -154,11 +154,28 @@ class SAM2StreamingProcessor:
'storage_device': self.device, # Keep everything on GPU 'storage_device': self.device, # Keep everything on GPU
'offload_video_to_cpu': False, 'offload_video_to_cpu': False,
'offload_state_to_cpu': False, 'offload_state_to_cpu': False,
# Add some required SAM2 internal structures # Add required SAM2 internal structures
'output_dict_per_obj': {}, 'output_dict_per_obj': {},
'temp_output_dict_per_obj': {}, 'temp_output_dict_per_obj': {},
'frames': None, # We provide frames manually 'frames': None, # We provide frames manually
'images': None, # We provide images manually 'images': None, # We provide images manually
# Additional SAM2 tracking fields
'frames_tracked_per_obj': {},
'obj_idx_to_id': {},
'obj_id_to_idx': {},
'click_inputs_per_obj': {},
'point_inputs_per_obj': {},
'mask_inputs_per_obj': {},
'output_dict': {},
'memory_bank': {},
'num_obj_tokens': 0,
'max_obj_ptr_num': 16, # SAM2 default
'multimask_output_in_sam': False,
'use_multimask_token_for_obj_ptr': True,
'max_inference_state_frames': -1, # No limit for streaming
'image_feature_cache': {},
'cached_features': {},
'consolidated_frame_inds': {},
} }
# Initialize some constants that SAM2 expects # Initialize some constants that SAM2 expects