search

jorge-pessoa / pytorch-msssim

PyTorch differentiable Multi-Scale Structural Similarity (MS-SSIM) loss
Other
454 stars 70 forks source link

Support for 3D images #10

Open rciszek opened 5 years ago

rciszek commented 5 years ago

The associated changes add support for 3D images. This is handy e.g. when dealing with MRI data.