yonigozlan HF Staff commited on
Commit
0ea4068
·
1 Parent(s): e905684
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -499,6 +499,8 @@ def propagate_masks(GLOBAL_STATE: gr.State):
499
  with torch.inference_mode():
500
  for frame_idx, frame in enumerate(GLOBAL_STATE.video_frames):
501
  pixel_values = None
 
 
502
  if inference_session.processed_frames is None or frame_idx not in inference_session.processed_frames:
503
  pixel_values = processor(images=frame, device="cuda", return_tensors="pt").pixel_values[0]
504
  sam2_video_output = model(inference_session=inference_session, frame=pixel_values, frame_idx=frame_idx)
 
499
  with torch.inference_mode():
500
  for frame_idx, frame in enumerate(GLOBAL_STATE.video_frames):
501
  pixel_values = None
502
+ print("inference_session.processed_frames", inference_session.processed_frames)
503
+ print("frame_idx", frame_idx)
504
  if inference_session.processed_frames is None or frame_idx not in inference_session.processed_frames:
505
  pixel_values = processor(images=frame, device="cuda", return_tensors="pt").pixel_values[0]
506
  sam2_video_output = model(inference_session=inference_session, frame=pixel_values, frame_idx=frame_idx)