Implementation of MSSSIM + L1 for Greyscale Images

VainF / pytorch-msssim

Fast and differentiable MS-SSIM and SSIM for pytorch.

MIT License

1.1k stars 126 forks source link

Implementation of MSSSIM + L1 for Greyscale Images #51

Open willxxy opened 9 months ago

willxxy commented 9 months ago

Hi, thanks for the great implementation. I am currently working with greyscale images and implemented the loss for 1 channel images here. Would you mind looking at the implementation and see if it is accurate?

Thank you.