diff --git a/inference_video_parallel.py b/inference_video_parallel.py index 00cdeaa..2ebf8b3 100644 --- a/inference_video_parallel.py +++ b/inference_video_parallel.py @@ -127,4 +127,5 @@ while success: buffer = [] _thread.start_new_thread(clear_buffer, (args, buffer)) pbar.close() -vid_out.release() +if not vid_out is None: + vid_out.release()