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 4 road parallel

Browse Source
This commit is contained in:
hzwer
2020-11-17 18:32:07 +08:00
parent 25dff5fb4e
commit f2977f9431
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
inference_mp4_4x_parallel.py
View File

@@ -87,8 +87,8 @@ while success:
mid2 = model.inference(mid1, I1)
mid0 = (((mid0 * 255.).cpu().detach().numpy().transpose(0, 2, 3, 1))).astype('uint8')
mid1 = (((mid1 * 255.).cpu().detach().numpy().transpose(0, 2, 3, 1))).astype('uint8')
mid2 = (((mid2* 255.).cpu().detach().numpy().transpose(0, 2, 3, 1))).astype('uint8')
writeframe(p, mid0, mid1, mid)
mid2 = (((mid2 * 255.).cpu().detach().numpy().transpose(0, 2, 3, 1))).astype('uint8')
writeframe(p, mid0, mid1, mid2)
pbar.update(4)
img_list = img_list[-1]
pbar.close()