nie-lang / DeepRectangling

CVPR2022 (Oral) - Deep Rectangling for Image Stitching: A Learning Baseline
229 stars 38 forks source link

test测试集上的可视化效果差的问题 #6

Open ChenJunzhi-buaa opened 2 years ago

ChenJunzhi-buaa commented 2 years ago

您好!

我按照你的步骤,下载好了您预训练好的模型,运行您的inference.py脚本,但是发现在您的519张测试集上的rectangling效果很差,无论是从生成的可视化结果来看,还是从量化指标来看。 请问这是为什么呢?

如下是量化指标,average ssim只有0.3. ===================Results Analysis================== average psnr: 9.62835509107355 average ssim: 0.31329677570284625

nie-lang commented 2 years ago

我试过直接把输入图片(拼接图)和gt求psnr/ssim,得到的结果都比这个高。你这可能是哪一步搞错了,建议在inference.py中输出一下你的输入图和mask图看看(输出的是实际读入网络里的图),检查是不是数据集读入有问题。

ChenJunzhi-buaa commented 2 years ago

十分感谢!我再排查一下。