[英]cv2 reading video frames causes python generator to delay output till the end
我有一个 python 生成器,它从视频中输出 cv2 图像(帧),并使用concurrent.futures.ThreadPoolExecutor
处理它。 我看到 tqdm 挂起,直到发电机用完为止。 但是,如果我从gen_frames_dummy
为执行程序提供数据,则 tqdm 会随着每个任务的完成而更新。
任何帮助将不胜感激。
def gen_frames(s):
for frame_num in frames_2_get:
# skip to frame in video
s.set(cv2.CAP_PROP_POS_FRAMES, frame_num-1)
res, frame = s.read()
if res:
yield frame
def gen_frames_dummy(s):
for frame_num in frames_2_get:
yield np.zeros((16,16,3))
def frame_op(f):
# process the frame
pass
with ThreadPoolExecutor() as executor:
frames_g = gen_frames(vid_capture)
list(tqdm(executor.map(frame_op, frames_g), total=len(frames2get))
map
等待所有工作人员完成,然后将结果传递给代码中的tqdm
。
我想您可以像这样重写以使tqdm
按预期工作。
for args in tqdm(frames_g, total=len(frames_2_get)):
executor.submit(frame_op, args)
声明:本站的技术帖子网页,遵循CC BY-SA 4.0协议,如果您需要转载,请注明本站网址或者原文地址。任何问题请咨询:yoyou2525@163.com.