github
/
arXiv2020-RIFE
mirror of https://github.com/hzwer/arXiv2020-RIFE
1
Fork 0
Code Releases Activity

Update inference_video.py

This commit is contained in:
hzwer 2021-06-23 12:19:04 +08:00 committed by GitHub
parent ace2911acb
commit b7b503dc69
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference_video.py
View File

@ -213,7 +213,7 @@ while True:
I1 = pad_image(I1)
I0_small = F.interpolate(I0, (32, 32), mode='bilinear', align_corners=False)
I1_small = F.interpolate(I1, (32, 32), mode='bilinear', align_corners=False)
ssim = ssim_matlab(I0_small, I1_small)
ssim = ssim_matlab(I0_small[:, :3], I1_small[:, :3])
if ssim > 0.995:
if skip_frame % 100 == 0: