Function msssim

utils/loss_utils.py:68–71

(rgb, gts)

ms_ssim = MultiScaleStructuralSimilarityIndexMeasure(data_range=1.0)

def msssim(rgb, gts):
    # assert (rgb.max() <= 1.05 and rgb.min() >= -0.05)
    # assert (gts.max() <= 1.05 and gts.min() >= -0.05)
    return ms_ssim(rgb, gts).item()