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)