怎么保存输出的图片

[windZhangAZ]

我使用放大两倍的功能 命令：

python main.py --dir_data cache/data --pretrain pretrained_model/IPT_sr2.pt --data_test Set5 --scale 2 --save cache/results/ipt --save_results --save_gt --test_only

结果[Set5 x2] PSNR: 38.391 (Best: 38.391 @epoch 1)
Forward: 116.60s
Saving...
Total: 117.04s
只有一个PSNR 在cache/results/ipt路径下什么也没有

[HantingChen]

目前在我的环境下可以正常保存，由于环境配置可能不完全一样，问题不太好定位。

可以尝试换个路径保存，或者尝试插入断点查看是否保存有问题，保存图片所用的function在：https://github.com/huawei-noah/Pretrained-IPT/blob/0700327c7bac18e13e322bb35e687d440e60a23a/utility.py#L151。

[windZhangAZ]

我在尝试各种功能时， 去模糊化的两个数据集 CBSD68 出现了问题 评估的时候找不到对应图片，但是Urban100 数据集没问题

我在想你们检索CBSD68图片是不是有特别规则 能不能在代码中找到对应？ 另外我通过你们给的链接下载的这个数据集名字是BSD300 你们是否对这个数据集有了修改

[HantingChen]

我在尝试各种功能时， 去模糊化的两个数据集 CBSD68 出现了问题 评估的时候找不到对应图片，但是Urban100 数据集没问题

我在想你们检索CBSD68图片是不是有特别规则 能不能在代码中找到对应？ 另外我通过你们给的链接下载的这个数据集名字是BSD300 你们是否对这个数据集有了修改

你好，相关代码已经更新，可以参考新的readme

Evaluation Process

Inference example:
For SR x2,x3,x4:

python main.py --dir_data $DATA_PATH --pretrain $MODEL_PATH --data_test Set5+Set14+B100+Urban100 --scale $SCALE

Note: path of images should be like: $MODEL_PATH/benchmark/Set5/HR/XXX.png and $MODEL_PATH/benchmark/Set5/LR_bicubic/XXX.png

For Denoise 30,50:

python main.py --dir_data $DATA_PATH --pretrain $MODEL_PATH --data_test CBSD68+Urban100 --scale 1 --denoise --sigma $NOISY_LEVEL

Note: path of images should be like: $MODEL_PATH/benchmark/CBSD68/XXX.png

For derain:

python main.py --dir_data $DATA_PATH --pretrain $MODEL_PATH --scale 1 --derain

Note: path of images should be like: $MODEL_PATH/Rain100L/rain-XXX.png and $MODEL_PATH/Rain100L/norain-XXX.png

[oneonejia]

想问一下读不到数据路径怎么办
在项目里新建了一个test_data文件夹，把benchmark、CBSD68、和Rain100L放在里面
输入python main.py --dir_data test_data/benchmark --pretrain model/IPT_sr2.pt --data_test Set5+Set14+B100+Urban100 --scale 1
python main.py --dir_data test_data/ --pretrain model/IPT_denoise30.pt --data_test CBSD68+Urban100 --scale 1 --denoise --sigma 30
python main.py --dir_data test_data/ --pretrain model/IPT_derain.pt --scale 1 --derain
都出现了0it [00:07, ?it/s]
[Urban100 x1] PSNR: nan (Best: nan @epoch 1)
类似与PSNR为nan的情况

[KeatsHao]

作者把cache 放在root目录里的😂😂