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.

Sign up for GitHub

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

Jump to bottom

SSIM During Training #28

Open

SuperBlueCat138123 opened this issue Feb 2, 2024 · 1 comment

SuperBlueCat138123 commented Feb 2, 2024

Hello,

I want to see the result of SSIM during training, how do I modify the train.py

Owner

zhengchen1999 commented Feb 4, 2024

You don't need to modify the code. Just add

    ssim:
      type: calculate_ssim
      crop_border: 2
      test_y_channel: True

after test_y_channel: True in yml.
Similar to the testing yml.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment