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)