SSIM calculate error

[stefenmax]

Hi, thanks for your work. When I want to reproduce the result using the DIV2K dataset. When calculate the ssim it raises the error
File "/mnt/new_ssd/stefen/code/AdaBM/src/utility.py", line 265, in ssim_matlab
(_, _, height, width) = img1.size()
ValueError: too many values to unpack (expected 4)
When I check the dimension of imag1, it has 5. Should I remove the first dimension? Thanks
(Pdb) img1.size()
torch.Size([1, 1, 3, 1336, 2020])

[stefenmax]

Plus, when I comment the ssim and continue to train 100 epochs use the following line
sh run.sh edsr 0 6 8
The average PSNR remain around 11.679, and the SR image looks not good.

[Cheeun]

Hi, thanks for your interest in our work!
Could you check which type of data you are using for training?
When I directly ran sh run.sh edsr 0 6 8, the data dimension was 4 instead of 5.

[stefenmax]

I used the DIV2K link you provided. It's PNG file

[Cheeun]

@stefenmax, sorry but I failed to reproduce your error.
Could you post the command that you used?