removed rfft2d init from GMM

lightly-ai · Oct 14, 2024 · 2c9e7c8 · 2c9e7c8
1 parent fa2fb17
commit 2c9e7c8
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 5 deletions.
diff --git a/lightly/transforms/gaussian_mixture_masks_transform.py b/lightly/transforms/gaussian_mixture_masks_transform.py
@@ -5,7 +5,6 @@
 from torch import Tensor
 
 from lightly.transforms.irfft2d_transform import IRFFT2DTransform
-from lightly.transforms.rfft2d_transform import RFFT2DTransform
 
 
 class GaussianMixtureMask:
@@ -29,9 +28,6 @@ def __init__(
             num_gaussians: Number of Gaussian kernels to generate in the mixture mask.
             std_range: Tuple containing the minimum and maximum standard deviation for the Gaussians.
         """
-
-        self.rfft2d_transform = RFFT2DTransform()
-
         self.num_gaussians = num_gaussians
         self.std_range = std_range
 

diff --git a/tests/transforms/test_gaussian_mixture_masks.py b/tests/transforms/test_gaussian_mixture_masks.py
@@ -5,6 +5,6 @@
 
 def test() -> None:
     transform = GaussianMixtureMask(20, (10, 15))
-    image = torch.rand(3, 32, 32)
+    image = torch.rand(3, 32, 17)
     output = transform(image)
     assert output.shape == (3, 32, 32)