search

zhengchen1999 / DAT

PyTorch code for our ICCV 2023 paper "Dual Aggregation Transformer for Image Super-Resolution"
Apache License 2.0
350 stars 27 forks source link

SSIM During Training #28

Open SuperBlueCat138123 opened 5 months ago

SuperBlueCat138123 commented 5 months ago

Hello,

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

zhengchen1999 commented 5 months ago

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.