fix something
This commit is contained in:
@@ -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,
|
||||||
|
|||||||
Reference in New Issue
Block a user