Update README

This commit is contained in:
hzwer
2020-11-26 18:47:55 +08:00
parent d3f99506b1
commit 8715002aab
2 changed files with 3 additions and 3 deletions

View File

@@ -93,8 +93,8 @@ We will release our training and benchmark validation code soon.
**Vimeo90K**
Download [Vimeo90K dataset](http://toflow.csail.mit.edu/) at ./vimeo_interp_test
```
$ python3 Vimeo90K_benchmark.py
(You will get 35.695PSNR and 0.9788SSIM)
$ python3 benchmark/Vimeo90K_benchmark.py
(Final result: "Avg PSNR: 35.695 SSIM: 0.9788")
```
## Citation

View File

@@ -37,4 +37,4 @@ for i in f:
psnr = -10 * math.log10(((I1 - mid) * (I1 - mid)).mean())
psnr_list.append(psnr)
ssim_list.append(ssim)
print(np.mean(psnr_list), np.mean(ssim_list))
print("Avg PSNR: {} SSIM: {}".format(np.mean(psnr_list), np.mean(ssim_list)))