caiyuanhao1998 / Retinexformer

"Retinexformer: One-stage Retinex-based Transformer for Low-light Image Enhancement" (ICCV 2023) & (NTIRE 2024 Challenge)

https://arxiv.org/abs/2303.06705

MIT License

828 stars 64 forks source link

关于SSIM #54

Closed Lost-Sunset closed 5 months ago

Lost-Sunset commented 5 months ago

（初学者同时也在研究毕设）运行了代码输出了psnr，如果想要获取或者输出SSIM该如何改动代码？相关代码1： if best_metric['psnr'] < current_metric: best_metric['psnr'] = current_metric

save best model

                best_metric['iter'] = current_iter
                model.save_best(best_metric)

相关代码2： best_metric = resume_state['best_metric'] best_psnr = best_metric['psnr'] best_iter = best_metric['iter'] logger.info(f'best psnr: {best_psnr} from iteration {best_iter}')

caiyuanhao1998 commented 5 months ago

Enhancement/test_from_dataset.py 里面有测SSIM的代码和已经封装好的函数，请参照那边

MoWangDaRen commented 2 weeks ago

你好请问您改好了吗