lichen/detecttracking
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add network image crop pipeline

Browse Source
This commit is contained in:
刘珈维
2025-01-14 13:38:17 +08:00
parent a16235a593
commit c7859a7792
1 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
stream_pipeline.py
View File

@ -16,6 +16,7 @@ from tracking import traclus as tr
from track_reid import parse_opt from track_reid import parse_opt
from track_reid import yolo_resnet_tracker from track_reid import yolo_resnet_tracker
from tracking.dotrack.dotracks_back import doBackTracks from tracking.dotrack.dotracks_back import doBackTracks
from PIL import Image
def save_event_subimgs(imgs, bboxes): def save_event_subimgs(imgs, bboxes):
img_list = {} img_list = {}
@ -97,9 +98,14 @@ def stream_pipeline(stream_dict):
event_tracks, video_frames = get_tracking_info(**parmDict) event_tracks, video_frames = get_tracking_info(**parmDict)
bboxes_opt = get_optimized_bboxes(event_tracks[0][1]) bboxes_opt = get_optimized_bboxes(event_tracks[0][1])
subimg_list = save_event_subimgs(video_frames, bboxes_opt) subimg_dict = save_event_subimgs(video_frames, bboxes_opt)
return subimg_list sub_images = []
for fid, img in subimg_dict.items():
pil_image = Image.fromarray(cv2.cvtColor(img, cv2.COLOR_BGR2RGB))
sub_images.append(pil_image)
return sub_images
def main(): def main():
''' '''