search

zhengchen1999 / DAT

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

How do I add something to code to calculate the SSIM? #63

Open lq5203344 opened 3 weeks ago

lq5203344 commented 3 weeks ago

Dear author: I want to calculate the SSIM on my own dataset in this DAT-S Model, but how do I add something to this current yaml file? And, whether I need to add some code for the train.py or other related part in this project.

zhengchen1999 commented 3 weeks ago

Hello, there's no need to modify the train.py file. You need to modify the path to your test dataset in the yml file.