← all repositories

Po-Hsun-Su/pytorch-ssim

A PyTorch implementation of the Structural Similarity Index (SSIM) as a differentiable loss function for image quality assessment.

1.9k stars Python Computer VisionML Frameworks
pytorch-ssim
Velocity · 7d
+0.6
★ / day
Trend
steady
star history

This library provides a pure PyTorch implementation of SSIM, a widely used image quality metric. It offers both functional (ssim) and module (SSIM class) interfaces for computing structural similarity between images. The implementation supports GPU acceleration via CUDA and integrates with PyTorch’s autograd system, enabling its use as a differentiable loss function in training loops for image generation, super-resolution, and enhancement models.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.