NHR dataset

[Harbinzzy]

您好，还有个问题想请教一下您，我使用的ITS的代码，用的您的nhr.pkl测试的结果，The average PSNR is 24.28 dB，The average SSIM is 0.96408 dB，与您论文中的不符。估计是我数据集搞错了吧。我是在这个仓库https://github.com/chaimi2013/3R 中的https://drive.google.com/drive/folders/1Q7Ep19eX3yBMwF4kfKI2TGUfeG-gXHKR 这个链接下载的NHR数据集。请问您能提供您论文中使用的NHR数据集吗？十分感谢您的帮助

[c-yn]

可以用这个试试https://drive.google.com/file/d/1ziEHGbrFnZeZGM_kuxoo4QSrQ63bfrC-/view?usp=drive_link

[Harbinzzy]

好的谢谢您

[c-yn]

哈喽，这两天比较忙没来得及回复。最后NHR测出来了么？

[Harbinzzy]

测出来了，十分感谢，麻烦再问一下，我看nhr数据集有的图片短边是小于256的，您的crop_size设置的是多少，batch_size=8学习率也是4e-4吗？感谢您的帮助 🙃 ***@***.***

[c-yn]

设置成256*256就可以。
参数如下：
parser.add_argument('--batch_size', type=int, default=8)
parser.add_argument('--learning_rate', type=float, default=1e-4)
parser.add_argument('--num_epoch', type=int, default=300)

[Harbinzzy]

十分感谢您的帮助

[Harbinzzy]

[rank0]: File "/data_share/zzy/FSNet_NHR/data/data_augment.py", line 34, in call
[rank0]: i, j, h, w = self.get_params(image, self.size)
[rank0]: File "/data_share/zzy/anaconda3/envs/torch/lib/python3.10/site-packages/torchvision/transforms/transforms.py", line 640, in get_params
[rank0]: raise ValueError(f"Required crop size {(th, tw)} is larger than input image size {(h, w)}")
[rank0]: ValueError: Required crop size (256, 256) is larger than input image size (169, 550)
有一些图片的短边小于256，无法直接crop，您是resize之后crop的吗

[c-yn]

我是把小的图片直接从训练集删除了，最后剩了16134张。