diff --git a/stream_pipeline.py b/stream_pipeline.py index 6bff7ce..c057a44 100644 --- a/stream_pipeline.py +++ b/stream_pipeline.py @@ -16,6 +16,7 @@ from tracking import traclus as tr from track_reid import parse_opt from track_reid import yolo_resnet_tracker from tracking.dotrack.dotracks_back import doBackTracks +from PIL import Image def save_event_subimgs(imgs, bboxes): img_list = {} @@ -97,9 +98,14 @@ def stream_pipeline(stream_dict): event_tracks, video_frames = get_tracking_info(**parmDict) 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(): '''