pytorch-msssim icon indicating copy to clipboard operation
pytorch-msssim copied to clipboard

Published 20 hours ago • verified publisher icon VainF

add mask support for SSIM

Open jinhwanlazy opened this issue 11 months ago • 0 comments

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

jinhwanlazy avatar Mar 12 '24 11:03 jinhwanlazy