Structural Similarity Index Measure (SSIM)

Image

Structural Similarity Index Measure (SSIM)

Module Interface

torchmetrics.StructuralSimilarityIndexMeasure

alias of _StructuralSimilarityIndexMeasure

Functional Interface

torchmetrics.functional.structural_similarity_index_measure(preds, target, gaussian_kernel=True, sigma=1.5, kernel_size=11, reduction='elementwise_mean', data_range=None, k1=0.01, k2=0.03, return_full_image=False, return_contrast_sensitivity=False)

Wrapper for deprecated import.

>>> import torch
>>> preds = torch.rand([3, 3, 256, 256])
>>> target = preds * 0.75
:rtype: :py:data:`~typing.Union`\[:py:class:`~torch.Tensor`, :py:data:`~typing.Tuple`\[:py:class:`~torch.Tensor`, :py:class:`~torch.Tensor`]]

>>> _structural_similarity_index_measure(preds, target)
tensor(0.9219)