From a826984ae3d41be534258f1320671c00376dfd5e Mon Sep 17 00:00:00 2001 From: zbchu2 Date: Tue, 9 Jul 2024 15:10:00 +0800 Subject: [PATCH] Fix video writing thread termination issue --- inference_video.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/inference_video.py b/inference_video.py index 468d66d..f445bd9 100644 --- a/inference_video.py +++ b/inference_video.py @@ -277,6 +277,9 @@ if args.montage: write_buffer.put(np.concatenate((lastframe, lastframe), 1)) else: write_buffer.put(lastframe) + +write_buffer.put(frame) + import time while(not write_buffer.empty()): time.sleep(0.1)