MIR.image_similarity.SSIMΒΆ

Structural Similarity Index (SSIM) utilities.

All SSIM functions use a Gaussian window with sigma = 1.5 and the constants C1 = 0.01^2 and C2 = 0.03^2, which assume intensities in [0, 1].

Functions

create_window(window_size, channel)

Create a 2D SSIM window tensor.

create_window_3D(window_size, channel)

Create a 3D SSIM window tensor.

gaussian(window_size, sigma)

Create a 1D Gaussian kernel for SSIM windows.

ssim(img1, img2[, window_size, size_average])

Compute 2D SSIM between two images (functional form of SSIM2D).

ssim3D(img1, img2[, window_size, size_average])

Compute 3D SSIM between two volumes.

Classes

SSIM2D(*args, **kwargs)

2D structural similarity (SSIM) between two images.

SSIM3D(*args, **kwargs)

3D SSIM loss between two volumes.