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 montage

Browse Source
This commit is contained in:
hzwer
2020-11-15 16:45:52 +08:00
parent a73928eebc
commit 8a50718b2a
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
inference_mp4_2x.py
View File

@@ -47,13 +47,13 @@ while success:
mid = model.inference(I0, I1)
mid = ((mid[0].cpu().detach().numpy().transpose(1, 2, 0))*255.).astype('uint8')
if args.montage:
output.write(torch.cat((lastframe, lastframe), 2))
output.write(torch.cat((lastframe, mid[:h, :w]), 2))
output.write(np.concatenate((lastframe, lastframe), 1))
output.write(np.concatenate((lastframe, mid[:h, :w]), 1))
else:
output.write(lastframe)
output.write(mid[:h, :w])
if args.montage:
output.write(torch.cat((frame, frame), 2))
output.write(np.concatenate((frame, frame), 1))
else:
output.write(frame)
output.release()