search

yuyi-sd / Robust_Rain_Removal

33 stars 5 forks source link

PSNR: nan #1

Open coordxyz opened 2 years ago

coordxyz commented 2 years ago

Hi, I got nan for evaluation. Any idea about this? Thx~

python robust.py --data_test RainHeavyTest --ext img --pre_train ../experiments/MPRNet_R_SEADD_MB_robust_pgd_Rain100H_e4/model/model_latest.pt --model MPRNet_R_SEADD_MB --test_only --save_results --save_gt --save_attack --save MPRNet_R_SEADD_MB_robust_pgd_test_Rain100H_e4_1 --n_GPUs 1 --attack_iters 20 --robust_epsilon 1 --robust_alpha 0.25

../data/test/small/ Making model... rest Loading model from ../experiments/MPRNet_R_SEADD_MB_robust_pgd_Rain100H_e4/model/model_latest.pt

Evaluation: 0it [00:00, ?it/s] [RainHeavyTest x2] PSNR: nan (Best: nan @epoch 1) [RainHeavyTest x2] sr PSNR: nan (Best: nan @epoch 1) [RainHeavyTest x2] lr PSNR: nan (Best: nan @epoch 1) Total time: 0.03s

yuyi-sd commented 2 years ago

I have a mistake on the data path. The current data are stored in './data/test/test/small/', where an extra '/test' appears here. You can rerange the data to './data/test/small', and I will fix it tomorrow!

Thank you for your kindly reminder!

coordxyz commented 2 years ago

Hi @yuyi-sd, Do you have any plan to share the code for training?

yuyi-sd commented 2 years ago

The training code is pretty messy. Leave your email here, I can send you one copy.

Xiaofeng-life commented 2 years ago

can I have your code of robust training? my email is cxf_svip@163.com.