Multi-Scale Structural Similarity

[jorge-pessoa]

Hello,

This SSIM implementation is a great addiction to the differentiable functions of PyTorch. However it is lacking the Multi-Scale Structural Similarity, a more commonly used metric in image assessment. I think it would be beneficial and relatively simple to implement and add this metric to this project.

I wrote a basic working MS-SSIM implementation based on SSIM from this project and using the Tensorflow implementation as reference. Would you consider adding the MS-SSIM metric if a pull request was sent?

Thank you,

[chaos5958]

Dear jorge-pessoa,

I also need MS-SSIM, how is your progress?
Could you share your version of MS-SSIM?

In addtion, I wonder whether there is GPU version of MS-SSIM regardless of using pytorch. (BUT SHOULD RUN ON PYTHON)

Thanks,
Hyunho

[jorge-pessoa]

Hello,

I'm currently using the mentioned MS-SSIM implementation on training with success. The results obtained are equal to this implementation https://github.com/tensorflow/models/blob/master/research/compression/image_encoder/msssim.py within +- 4 decimal places so i suppose it is mostly correct.

I forked this repository and included the MS-SSIM implementation which you can find at https://github.com/jorge-pessoa/pytorch-msssim. I hope it is of some use to you.

Thanks,
Jorge