This commit is contained in:
2025-07-27 09:23:15 -07:00
parent 7b3ffb7830
commit 9b7f36fec2

View File

@@ -154,11 +154,28 @@ class SAM2StreamingProcessor:
'storage_device': self.device, # Keep everything on GPU
'offload_video_to_cpu': False,
'offload_state_to_cpu': False,
# Add some required SAM2 internal structures
# Add required SAM2 internal structures
'output_dict_per_obj': {},
'temp_output_dict_per_obj': {},
'frames': None, # We provide frames 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