Add a missing : and imports

datasets/raise/fft/fourier_to_image.py

@@ -1,6 +1,7 @@
 from PIL import Image
 import numpy as np
 from scipy import fftpack
+import matplotlib.pyplot as plt
 
 fft = np.array(Image.open('fft.png').convert('L'))# / 255
 print(fft)

datasets/raise/fft/image_to_fourier.py

@@ -1,6 +1,7 @@
 from PIL import Image
 import numpy as np
 from scipy import fftpack
+import matplotlib.pyplot as plt
 
 image = np.array(Image.open('image_4.png').convert('L'))# / 255
 print(image)
@@ -13,10 +14,6 @@ height, width = fft1.shape
 for x in range(width):
     fft1[height // 2, x] = np.median(fft1[:, x])
 
-
-
-
-
 # save the image
 # np.log10
 plt.imshow(np.log10(abs(fft1)))

datasets/raise/utils.py

@@ -33,7 +33,7 @@ class Denoiser(Enum):
     def __str__(self):
         return self.name.lower()
 
-class Distance(Enum)
+class Distance(Enum):
     ROOT_MEAN_SQUARE = auto()
     PEARSON_CORRELATION = auto()