search

VainF / pytorch-msssim

Fast and differentiable MS-SSIM and SSIM for pytorch.
MIT License
1.1k stars 126 forks source link

add mask support for SSIM #53

Open jinhwanlazy opened 8 months ago

jinhwanlazy commented 8 months ago

ssim function accepts additional input, mask, which is boolean tensor of same size as the input images.

This PR closes https://github.com/VainF/pytorch-msssim/issues/50