3dgs
public
fused-ssim
v0.1.0
improved SSIM metric
Fully Fused Differentiable SSIM This repository contains an efficient fully-fused implementation of SSIM which is differentiable in nature. As per the original SSIM paper, this implementation uses 11x11 sized convolution kernel. The weights for it have been hardcoded and this is another reason for it's speed. This implementation currently only supports 2D images but with variable number of channels and batch size. This is used in 3DGS.
Install
pixiaddfused-ssim
micromambainstall-c https://repo.prefix.dev/3dgsfused-ssim
Version
Platforms
Last published