caiyuanhao1998 / MST-plus-plus

"MST++: Multi-stage Spectral-wise Transformer for Efficient Spectral Reconstruction" (CVPRW 2022) & (Winner of NTIRE 2022 Spectral Recovery Challenge) and a toolbox for spectral reconstruction

Home Page:https://arxiv.org/abs/2204.07908

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

请问为什么计算评价指标,只使用图像的中间区域计算呢?

yanxinpeng517 opened this issue · comments

loss_mrae = criterion_mrae(output[:, :, 128:-128, 128:-128], target[:, :, 128:-128, 128:-128])
loss_rmse = criterion_rmse(output[:, :, 128:-128, 128:-128], target[:, :, 128:-128, 128:-128])
loss_psnr = criterion_psnr(output[:, :, 128:-128, 128:-128], target[:, :, 128:-128, 128:-128])
而不是用
loss_mrae = criterion_mrae(output, target)
loss_rmse = criterion_rmse(output, target)
loss_psnr = criterion_psnr(output, target)

这个是 NTIRE val 数据集规定的计算区域。当然,你也可以改成整张图像。
当时我们都是直接提交mat,然后服务器端算出性能指标。
然后服务器里面运行的代码,就是指定这一区域。

好的谢谢