计算PSNR时报错 RuntimeWarning: divide by zero encountered in double_scalars

报错的原因是使用skimage库计算PSNR时,MSE为0作为分母导致的错误。

解决办法:找到skimage库中报错的代码位置,将163行代码:

return 10 * np.log10((data_range ** 2) / err)

改为:

if err < 1.0e-10:

    return 100

else:

    return 10 * np.log10((data_range ** 2) / err)

你可能感兴趣的:(python)