Mirrors/flowframes
1
0
Fork 0
mirror of https://github.com/n00mkrad/flowframes.git synced 2026-07-13 14:00:48 +02:00
Code Issues Packages Projects Releases Wiki Activity

rife-cuda pkg update: Fixed imgformat arg, decreased buffer size

Browse Source
This commit is contained in:
N00MKRAD
2020-12-10 17:47:08 +01:00
parent f82797154b
commit 97f1548841
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
Pkgs/rife-cuda/inference_video.py
View File

@@ -96,8 +96,8 @@ def clear_write_buffer(user_args, write_buffer):
if item is None:
break
if user_args.png:
print('=> {:0>8d}.png'.format(cnt))
cv2.imwrite('{}/{:0>8d}.png'.format(interp_output_path, cnt), item[:, :, ::-1])
print('=> {:0>8d}.{}'.format(cnt, args.imgformat))
cv2.imwrite('{}/{:0>8d}.{}'.format(interp_output_path, cnt, args.imgformat), item[:, :, ::-1])
#cv2.imwrite('vid_out/{:0>7d}.png'.format(cnt), item[:, :, ::-1])
cnt += 1
else:
@@ -132,8 +132,8 @@ skip_frame = 1
if args.montage:
lastframe = lastframe[:, left: left + w]
write_buffer = Queue(maxsize=500)
read_buffer = Queue(maxsize=500)
write_buffer = Queue(maxsize=200)
read_buffer = Queue(maxsize=200)
_thread.start_new_thread(build_read_buffer, (args, read_buffer, videogen))
_thread.start_new_thread(clear_write_buffer, (args, write_buffer))