Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2026-02-24 04:19:41 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix timeout

Browse Source
This commit is contained in:
hzwer
2020-11-23 11:54:25 +08:00
parent 4d86885867
commit d6c535cc54
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
inference_video.py
View File

@@ -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()

5
inference_video_parallel.py
View File

@@ -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()