test

src/benchmark/__init__.py

@@ -0,0 +1,13 @@
+from torchmetrics.image import (
+    LearnedPerceptualImagePatchSimilarity,
+    PeakSignalNoiseRatio,
+    StructuralSimilarityIndexMeasure,
+)
+
+psnr = PeakSignalNoiseRatio(255.0, reduction="none")
+ssim = StructuralSimilarityIndexMeasure(reduction="none")
+lpips = LearnedPerceptualImagePatchSimilarity(net_type="alex", reduction="none")
+
+
+def benchmark(image1, image2):
+    return psnr(image1, image2), ssim(image1, image2), lpips(image1, image2)