Spaces:
Runtime error
Runtime error
Commit
·
0ea4068
1
Parent(s):
e905684
debug
Browse files
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)
|