Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Regarding Testing Metrics #17

Open
liu645liu opened this issue Oct 24, 2024 · 6 comments
Open

Regarding Testing Metrics #17

liu645liu opened this issue Oct 24, 2024 · 6 comments

Comments

@liu645liu
Copy link

Hello, thank you for your excellent work. I used your pre-trained model to test test.py, and I only get the PSNR metric on the three test sets, without any SSIM results. Which part should I modify?

@sunshangquan
Copy link
Owner

Hi @liu645liu , you could use compute_psnr.py for computing metrics.

@liu645liu
Copy link
Author

Thank you for your reply. I have run compute_psnr.py but the PSNR it got is very different from the result run using test.py. For example, on the Raindrop test set, the PSNR obtained using test.py is 33.02 but the result obtained using compute_psnr.py is 27.13, why is this?

@sunshangquan
Copy link
Owner

Could you make sure the results are based on test_a, instead of test_b?

@liu645liu
Copy link
Author

Yes, and not only the Raindrop test set, but also the test results of several other data sets are much worse. I am thinking whether compute_psnr.py did not use your pre-trained model for testing, but I have modified the relevant configuration and used it. best.pth, compute_psnr.py should have the same effect as test.py. Can you give me an answer?
ce155f3559e9288613228d4f9e57eda1
9b3b77bada6c43dc22b57ebbfcfc648d

@sunshangquan
Copy link
Owner

Your second figure implies that you are testing the psnr and ssim between low-quality and ground-truth, rather than restored images and ground-truth. Could you make sure this?

@sunshangquan
Copy link
Owner

You could use test_histoformer.py for generating restored images.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants