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

Kill warning

Browse Source
This commit is contained in:
hzwer
2020-11-15 16:31:33 +08:00
parent 2d3f6283a5
commit 0a0c10052b
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
inference_mp4_2x.py
View File

@@ -36,8 +36,8 @@ while success:
I1 = torch.from_numpy(np.transpose(frame, (2,0,1)).astype("float32") / 255.).to(device).unsqueeze(0)
I0 = F.pad(I0, padding)
I1 = F.pad(I1, padding)
if (F.interpolate(I0, (16, 16), mode='bilinear')
- F.interpolate(I1, (16, 16), mode='bilinear')).abs().mean() > 0.2:
if (F.interpolate(I0, (16, 16), mode='bilinear', recompute_scale_factor=False)
- F.interpolate(I1, (16, 16), mode='bilinear', recompute_scale_factor=False)).abs().mean() > 0.2:
mid = lastframe
else:
mid = model.inference(I0, I1)