Adding missing SSIM example.

gugarosa · Dec 23, 2020 · 25c87bc · 25c87bc
1 parent 371f723
commit 25c87bc
Showing 1 changed file with 13 additions and 0 deletions.
diff --git a/examples/math/calculate_ssim.py b/examples/math/calculate_ssim.py
@@ -0,0 +1,13 @@
+import torch
+
+import learnergy.math.metrics as m
+
+# Defines two random tensors
+v = torch.normal(0, 1, size=(10, 784))
+x = torch.normal(0, 1, size=(10, 28, 28))
+
+# Calculates the SSIM between them
+ssim = m.calculate_ssim(v, x)
+
+# Prints the result
+print(ssim)