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 bugs

Browse Source
This commit is contained in:
hzwer
2020-11-16 17:37:50 +08:00
parent f37f971a72
commit 61a48ef61c
2 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
inference_mp4_2x.py
View File

@@ -60,7 +60,8 @@ while success:
p = (F.interpolate(I0, (16, 16), mode='bilinear', align_corners=False)
- F.interpolate(I1, (16, 16), mode='bilinear', align_corners=False)).abs().mean()
if p < 1e-3 and args.skip:
print("Warning: Your video has {} static frames, skipping them change the duration of the generated video.".format(cnt))
if cnt % 10 == 0:
print("Warning: Your video has {} static frames, skipping them change the duration of the generated video.".format(cnt))
cnt += 1
pbar.update(1)
continue
@@ -74,9 +75,7 @@ while success:
output.write(np.concatenate((lastframe, mid1[:h, :w]), 1))
else:
output.write(lastframe)
output.write(mid0[:h, :w])
output.write(mid1[:h, :w])
output.write(mid2[:h, :w])
pbar.update(1)
if args.montage:
output.write(np.concatenate((lastframe, lastframe), 1))

3
inference_mp4_4x.py
View File

@@ -60,7 +60,8 @@ while success:
p = (F.interpolate(I0, (16, 16), mode='bilinear', align_corners=False)
- F.interpolate(I1, (16, 16), mode='bilinear', align_corners=False)).abs().mean()
if p < 1e-3 and args.skip:
print("Warning: Your video has {} static frames, skipping them may change the duration of the generated video.".format(cnt))
if cnt % 10 == 0:
print("Warning: Your video has {} static frames, skipping them may change the duration of the generated video.".format(cnt))
cnt += 1
pbar.update(1)
continue