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

another approach for final decoder

Browse Source
This commit is contained in:
neulus
2025-09-29 23:20:27 +09:00
parent 12a165e461
commit 0ccf1ff42d
14 changed files with 88 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
src/benchmark/__init__.py
View File

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