We used an implementation from https://github.com/Po-Hsun-Su/pytorch-ssim to compute ssim loss.
Please place pytorch_ssim/__init__.py into the folder ./pytorch_ssim/
.
We used an implementation from https://github.com/Po-Hsun-Su/pytorch-ssim to compute ssim loss.
Please place pytorch_ssim/__init__.py into the folder ./pytorch_ssim/
.