Understanding ssim

Hi man, great job! I'm trying to understand your code by following the paper. However I didn't get what is the purpose of SSIM class at all. Please, could you gimme some tip about it? Maybe by point out which part of the paper is related to this module. Thanks!