From d6c535cc540bcd490e281a116bf17bb1b001cff2 Mon Sep 17 00:00:00 2001
From: hzwer <598460606@163.com>
Date: Mon, 23 Nov 2020 11:54:25 +0800
Subject: [PATCH] Fix timeout

---
 inference_video.py          | 5 ++++-
 inference_video_parallel.py | 5 ++++-
 2 files changed, 8 insertions(+), 2 deletions(-)

diff --git a/inference_video.py b/inference_video.py
index 74bd1d5..2925834 100644
--- a/inference_video.py
+++ b/inference_video.py
@@ -55,7 +55,7 @@ else:
 def clear_buffer(user_args, buffer):
     cnt = 0
     while True:
-        item = buffer.get(timeout=1)
+        item = buffer.get()
         if item is None:
             break
         if user_args.png:
@@ -124,6 +124,9 @@ if args.montage:
     buffer.put(np.concatenate((lastframe, lastframe), 1))
 else:
     buffer.put(lastframe)
+import time
+while(not buffer.empty()):
+    time.sleep(0.1)
 pbar.close()
 if not vid_out is None:
     vid_out.release()
diff --git a/inference_video_parallel.py b/inference_video_parallel.py
index 379ccb3..e6adb3f 100644
--- a/inference_video_parallel.py
+++ b/inference_video_parallel.py
@@ -82,7 +82,7 @@ def write_frame(i0, infs, i1, p, user_args):
 def clear_buffer(user_args, buffer):    
     global cnt
     while True:
-        item = buffer.get(timeout=1)
+        item = buffer.get()
         if item is None:
             break
         if user_args.png:
@@ -128,6 +128,9 @@ for frame in videogen:
         pbar.update(4)
         img_list = img_list[-1:]
 buffer.put(img_list[0])
+import time
+while(not buffer.empty()):
+    time.sleep(0.1)
 pbar.close()
 if not vid_out is None:
     vid_out.release()