模型验证问题

[Lxh1992]

作者，您好！ 我在使用预训练的“PSNR4131_SSIM9945.pth”在ITS数据集上进行验证时发现以下问题，因为在SOTS-Indoor数据集中清晰图像会有一圈白边，导致图像尺寸为(1,3,480,640)，如果不去白边，运行eval.py会出现以下问题 evaluation: 0%| | 0/500 [00:01<?, ?it/s] Traceback (most recent call last): File "eval.py", line 59, in <module> avg_psnr, avg_ssim = eval(val_loader, network) File "eval.py", line 35, in eval psnr_tmp = val_psnr(output, clear_img) File "E:\Mycode\DEA-Net\code\utils\metric.py", line 53, in val_psnr imdff = pred - gt ValueError: operands could not be broadcast together with shapes (1,3,460,620) (1,3,480,640) 但是如果把白边去除，让清晰图像的尺寸为 (1,3,460,620)时，验证结果如下： evaluation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 500/500 [00:29<00:00, 16.89it/s] Evaluation on ITS PSNR:25.27524749248922 SSIM:0.8369731739759445 远低于41.31和0.9945 几种图像如下所示 不知道是哪边设置有问题，希望作者能帮忙指点一下，十分感谢！ 相关图像如下图所示： 原始清晰图像 去除白边后的图像 有雾图像

[weiMytian]

作者，您好！ 我在使用预训练的“PSNR4131_SSIM9945.pth”在ITS数据集上进行验证时发现以下问题，因为在SOTS-Indoor数据集中清晰图像会有一圈白边，导致图像尺寸为(1,3,480,640)，如果不去白边，运行eval.py会出现以下问题 evaluation: 0%| | 0/500 [00:01<?, ?it/s] Traceback (most recent call last): File "eval.py", line 59, in <module> avg_psnr, avg_ssim = eval(val_loader, network) File "eval.py", line 35, in eval psnr_tmp = val_psnr(output, clear_img) File "E:\Mycode\DEA-Net\code\utils\metric.py", line 53, in val_psnr imdff = pred - gt ValueError: operands could not be broadcast together with shapes (1,3,460,620) (1,3,480,640) 但是如果把白边去除，让清晰图像的尺寸为 (1,3,460,620)时，验证结果如下： evaluation: 100%|███████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████████| 500/500 [00:29<00:00, 16.89it/s] Evaluation on ITS PSNR:25.27524749248922 SSIM:0.8369731739759445 远低于41.31和0.9945 几种图像如下所示 不知道是哪边设置有问题，希望作者能帮忙指点一下，十分感谢！ 相关图像如下图所示： 原始清晰图像 去除白边后的图像 有雾图像

你好，我不是作者哈，我做的实验去掉白边后验证的结果和作者论文给的结果是一致的，你可以看一下自己裁剪方式是否有问题。我复现的问题是这个权重去验证ITS数据集的val数据集的时候差距较大。

[ChengchengFU]

clear图像中心裁剪后再使用（(480,640)-->(460,620)），验证无问题

[Lxh1992]

感谢感谢，终于解决了！