llm_agent

This commit is contained in:
2025-04-15 09:26:24 +08:00
parent ad850221c5
commit 9400ae904a
25 changed files with 52650 additions and 39 deletions

View File

@ -37,7 +37,8 @@ def get_optimized_bboxes(event_tracks):
points.append([int(ele[2]), int(ele[3])])
labels.append(int(ele[4])) # track_id
points = np.array(points)
if len(points) == 0:
return []
partitions, indices = tr.partition(points, progress_bar=False, w_perpendicular=100, w_angular=10)
bboxes_opt = []
@ -98,6 +99,8 @@ def stream_pipeline(stream_dict):
event_tracks, video_frames = get_tracking_info(**parmDict)
bboxes_opt = get_optimized_bboxes(event_tracks[0][1])
if len(bboxes_opt) == 0:
return []
subimg_dict = save_event_subimgs(video_frames, bboxes_opt)
sub_images = []