赞
踩
在进行超分辨率重建后想计算SSIM和PSNR,最开始发现导入compare_psnr,compare_ssim居然报错了,
上网一查发现版本更新换成了structural_similarity和peak_signal_noise_ratio。
解决
之后又发现
好家伙,这么一大段话,我估计 multichannel=True也已经不用了,报错中说
set channel_axis to the axis number corresponding to the channels
将你的通道维度告诉channel_axis
解决
续:后来我完成了我的模型训练等等一切,最后计算全部数据的SSIM和PSNR时的代码我会放在下一篇文章中,但在这时又遇到了一个巨坑的错。
字面翻译:ValueError:image_true的强度值超出了其数据类型的预期范围。请手动指定data_range。
由于我的fake_img和hr_img是经过transforms.ToTensor()出来的,因此他们都在[0,1]的范围内,只用用一个data_range值为255就ok了,虽然我并不是很理解,但是代码就这么通了 0。0
修改前
修改后
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。