add network image crop pipeline
This commit is contained in:
@ -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():
|
||||
'''
|
||||
|
Reference in New Issue
Block a user