search

sanghyun-son / EDSR-PyTorch

PyTorch version of the paper 'Enhanced Deep Residual Networks for Single Image Super-Resolution' (CVPRW 2017)
MIT License
2.42k stars 668 forks source link

Only PSNR, missing SSIM #283

Open baobaovuongcho opened 3 years ago

baobaovuongcho commented 3 years ago

Hello, I want to calculate both PSNR/SSIM however this code was not provided to calculate SSIM as the public paper. How can I calculate them, Thanks.

HolmesShuan commented 3 years ago

An unofficial PyTorch implementation https://github.com/HolmesShuan/EDSR-ssim.

lidanyang916 commented 3 years ago

Hello, I want to calculate both PSNR/SSIM however this code was not provided to calculate SSIM as the public paper. How can I calculate them, Thanks.

Did you find a solution? I have the same problem.

wwang0107 commented 2 years ago

An unofficial PyTorch implementation https://github.com/HolmesShuan/EDSR-ssim.

Should i just add the ssim function to the utility.py or replace the psnr function, train it again?