Something wrong with calculate_ssim?

xinntao / EDVR

Winning Solution in NTIRE19 Challenges on Video Restoration and Enhancement (CVPR19 Workshops) - Video Restoration with Enhanced Deformable Convolutional Networks. EDVR has been merged into BasicSR and this repo is a mirror of BasicSR.

https://github.com/xinntao/BasicSR

1.49k stars 318 forks source link

Something wrong with calculate_ssim? #148

Closed ryanxingql closed 4 years ago

ryanxingql commented 4 years ago

metrics.py -> def calculate_ssim

if img1.shape[2] == 3:
    ssims = []
    for i in range(3):
        ssims.append(ssim(img1, img2))

Should be:

ssims.append(ssim(np.squeeze(img1[:,:,i]), np.squeeze(img2[:,:,i])))