Mirrors/ECCV2022-RIFE
1
0
Fork 0
mirror of https://github.com/hzwer/ECCV2022-RIFE.git synced 2026-02-23 20:09:40 +01:00
Code Issues Packages Projects Releases Wiki Activity

Fix clear

Browse Source
This commit is contained in:
hzwer
2020-11-22 10:35:35 +08:00
parent 9ea44e7e7b
commit 4b921c898b
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
inference_video.py
View File

@@ -74,6 +74,7 @@ skip_frame = 1
if args.montage:
frame = frame[:, left: left + w]
buffer = Queue()
_thread.start_new_thread(clear_buffer, (args, buffer))
while success:
lastframe = frame
success, frame = videoCapture.read()
@@ -119,14 +120,10 @@ while success:
if args.exp == 4:
buffer.put(mid2[:h, :w])
pbar.update(1)
if buffer.qsize() > 100:
_thread.start_new_thread(clear_buffer, (args, buffer))
buffer.clear()
if args.montage:
buffer.put(np.concatenate((lastframe, lastframe), 1))
else:
buffer.put(lastframe)
_thread.start_new_thread(clear_buffer, (args, buffer))
pbar.close()
if not vid_out is None:
vid_out.release()