/DiffSSIM.jl

SSIM

Primary LanguageJulia

Differentiable SSIM loss function

Uses Flux Convolution layer instead of ImageQualityIndexes.jl library for GPU support.

Loss function is defiined and an example training steps are described.

This is not a library but a reference for SSIM implementation.