PSNR, SSIM calculation

[ryanlu2240]

Hi, thanks for such a convenient evaluation tool! I want to ask about the calculation for PSNR and SSIM. Is it calculated on the Y channel in the YCbCr color space or on RGB channels?

[chaofengc]

Thanks to your interest! By default, SSIM uses YCbCr space and PSNR uses RGB space. Alternatively, you can pass the create_metric('psnr', test_y_channel=True) to control it.