Enhance MultiScaleSSIM: Add data_range property and update test fixtures for larger image sizes #17

Author: selimfirat

kaira/metrics/image/ssim.py

@@ -227,6 +227,11 @@ def compute_with_stats(self, x: torch.Tensor, y: torch.Tensor, *args: Any, **kwa
             return values.mean(), torch.tensor(0.0)
         return values.mean(), values.std()
 
+    @property
+    def data_range(self) -> float:
+        """Get the data range used by the underlying torchmetrics implementation."""
+        return self.ms_ssim.data_range
+
 
 # Alias for backward compatibility
 SSIM = StructuralSimilarityIndexMeasure

tests/losses/test_losses_image.py

@@ -26,13 +26,13 @@
 @pytest.fixture
 def sample_images():
     """Fixture for creating sample image tensor."""
-    return torch.rand(2, 3, 32, 32)
+    return torch.rand(2, 3, 192, 192)  # Larger size for MS-SSIM compatibility
 
 
 @pytest.fixture
 def sample_target_images():
     """Fixture for creating sample target image tensor."""
-    return torch.rand(2, 3, 32, 32)
+    return torch.rand(2, 3, 192, 192)  # Larger size for MS-SSIM compatibility
 
 
 def test_mse_loss(sample_images, sample_target_images):