neulus/cloud-removal
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix wrong average of psnr

Browse Source
This commit is contained in:
neulus
2025-10-02 19:40:00 +09:00
parent a601dc6095
commit 6bb6c09638
17 changed files with 221 additions and 39 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/benchmark/__init__.py
View File

@@ -4,6 +4,7 @@ from torchmetrics.image import (
StructuralSimilarityIndexMeasure,
)
psnr = PeakSignalNoiseRatio(1.0, reduction="none", dim=(1, 2, 3))
ssim = StructuralSimilarityIndexMeasure(data_range=1.0, reduction="none")
lpips = LearnedPerceptualImagePatchSimilarity(
net_type="alex", reduction="none", normalize=True
@@ -11,5 +12,4 @@ lpips = LearnedPerceptualImagePatchSimilarity(
def benchmark(image1, image2):
psnr = PeakSignalNoiseRatio(1.0, reduction="none")
return psnr(image1, image2), ssim(image1, image2), lpips(image1, image2)