Spaces:
Sleeping
Sleeping
Zhen Ye Claude Opus 4.6 commited on
Commit ·
fb09dce
1
Parent(s): 8c60c06
feat(isr): store latest frame in writer thread for ISR cropping
Browse files- inference.py +3 -1
inference.py
CHANGED
|
@@ -21,7 +21,7 @@ from models.model_loader import load_detector, load_detector_on_device
|
|
| 21 |
from models.segmenters.model_loader import load_segmenter, load_segmenter_on_device
|
| 22 |
from models.depth_estimators.model_loader import load_depth_estimator, load_depth_estimator_on_device
|
| 23 |
from utils.video import StreamingVideoWriter
|
| 24 |
-
from jobs.storage import set_track_data
|
| 25 |
import tempfile
|
| 26 |
import json as json_module
|
| 27 |
|
|
@@ -946,6 +946,7 @@ def run_inference(
|
|
| 946 |
# Store tracks for frontend access
|
| 947 |
if job_id:
|
| 948 |
set_track_data(job_id, next_idx, dets)
|
|
|
|
| 949 |
|
| 950 |
next_idx += 1
|
| 951 |
|
|
@@ -1314,6 +1315,7 @@ def run_grounded_sam2_tracking(
|
|
| 1314 |
# Store tracks for frontend
|
| 1315 |
if job_id:
|
| 1316 |
set_track_data(job_id, next_idx, copy.deepcopy(dets))
|
|
|
|
| 1317 |
else:
|
| 1318 |
if job_id:
|
| 1319 |
set_track_data(job_id, next_idx, [])
|
|
|
|
| 21 |
from models.segmenters.model_loader import load_segmenter, load_segmenter_on_device
|
| 22 |
from models.depth_estimators.model_loader import load_depth_estimator, load_depth_estimator_on_device
|
| 23 |
from utils.video import StreamingVideoWriter
|
| 24 |
+
from jobs.storage import set_track_data, store_latest_frame
|
| 25 |
import tempfile
|
| 26 |
import json as json_module
|
| 27 |
|
|
|
|
| 946 |
# Store tracks for frontend access
|
| 947 |
if job_id:
|
| 948 |
set_track_data(job_id, next_idx, dets)
|
| 949 |
+
store_latest_frame(job_id, p_frame)
|
| 950 |
|
| 951 |
next_idx += 1
|
| 952 |
|
|
|
|
| 1315 |
# Store tracks for frontend
|
| 1316 |
if job_id:
|
| 1317 |
set_track_data(job_id, next_idx, copy.deepcopy(dets))
|
| 1318 |
+
store_latest_frame(job_id, frm)
|
| 1319 |
else:
|
| 1320 |
if job_id:
|
| 1321 |
set_track_data(job_id, next_idx, [])
|