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

Use CUDA to evaluation

Browse Source
This commit is contained in:
hzwer
2020-11-15 13:18:09 +08:00
parent c3073bbfcf
commit ab99580bd9
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Vimeo90K_benchmark.py
View File

@@ -33,7 +33,7 @@ for i in f:
mid = np.round((mid * 255).cpu().numpy()).astype('uint8').transpose(1, 2, 0) / 255.
I1 = I1 / 255.
psnr = -10 * math.log10(((I1 - mid) * (I1 - mid)).mean())
ssim = ssim_matlab(torch.tensor(I1).unsqueeze(0).float(), torch.tensor(mid).unsqueeze(0).float())
ssim = ssim_matlab(torch.tensor(I1).unsqueeze(0).float().to(device), torch.tensor(mid).unsqueeze(0).float().to(device)).cpu().numpy()
psnr_list.append(psnr)
ssim_list.append(ssim)
print(np.mean(psnr_list), np.mean(ssim_list))