llm_agent
This commit is contained in:
@ -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 = []
|
||||
|
Reference in New Issue
Block a user