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

pixel shuffle

Browse Source
This commit is contained in:
neulus
2025-10-01 16:30:05 +09:00
parent 8966cafb8f
commit 49025c4d87
19 changed files with 162 additions and 101 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/benchmark/__init__.py
View File

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