From 7f78048d89edfeb65505a7894b8cdae7488f50fc Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Fri, 20 Nov 2020 21:07:26 +0800
Subject: [PATCH] Update inference_video.py

---
 inference_video.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/inference_video.py b/inference_video.py
index e0955ac..1894571 100644
--- a/inference_video.py
+++ b/inference_video.py
@@ -124,4 +124,5 @@ else:
     buffer.append(lastframe)
 _thread.start_new_thread(clear_buffer, (args, buffer))
 pbar.close()
-vid_out.release()
+if not vid_out is None:
+    vid_out.release()