fix something

This commit is contained in:
2025-07-27 08:58:43 -07:00
parent abc48604a1
commit 3a59e87f3e

View File

@@ -144,7 +144,7 @@ class SAM2StreamingProcessor:
'temp_output_dict_per_obj': {}, 'temp_output_dict_per_obj': {},
'consolidated_frame_inds': {}, 'consolidated_frame_inds': {},
'tracking_has_started': False, 'tracking_has_started': False,
'num_frames': video_info['frame_count'], 'num_frames': video_info.get('total_frames', video_info.get('frame_count', 0)),
'video_height': video_info['height'], 'video_height': video_info['height'],
'video_width': video_info['width'], 'video_width': video_info['width'],
'device': self.device, 'device': self.device,