search

huawei-noah / Pretrained-IPT

Apache License 2.0
424 stars 63 forks source link

怎么保存输出的图片 #28

Open windZhangAZ opened 2 years ago

windZhangAZ commented 2 years ago

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

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 commented 2 years ago

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

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

windZhangAZ commented 2 years ago

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

空值

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

HantingChen commented 2 years ago

我在尝试各种功能时， 去模糊化的两个数据集 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 commented 2 years ago

想问一下读不到数据路径怎么办 在项目里新建了一个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 commented 2 years ago

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