add network image crop pipeline

This commit is contained in:
2025-01-14 13:38:17 +08:00
parent a16235a593
commit c7859a7792

View File

@ -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():
'''