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
Files changed (1) hide show
  1. 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, [])