python: psnr和ssim计算

最新推荐文章于 2024-03-03 20:52:34 发布

tony365

最新推荐文章于 2024-03-03 20:52:34 发布

阅读量358

点赞数

文章标签： python 计算机视觉 opencv

本文链接：https://blog.csdn.net/tywwwww/article/details/131781354

版权

psnr 和 ssim计算

from skimage.metrics import peak_signal_noise_ratio as compare_psnr
from skimage.metrics import structural_similarity as compare_ssim

denoised = np.clip(denoised * 255.0 + 0.5, 0, 255).astype(np.uint8)

temp_psnr = compare_psnr(denoised, gt, data_range=255)
temp_ssim = compare_ssim(denoised, gt, data_range=255, channel_axis=-1)

cv2.imwrite(os.path.join(save_dir, '%s_%.2f_out.png' % (filename, temp_psnr)), denoised)
cv2.imwrite(os.path.join(save_dir, '%s_NOISY.png' % (filename)), noisy)
cv2.imwrite(os.path.join(save_dir, '%s_GT.png' % (filename)), gt)