From 337a6eca8488309ffe6743d8a54091729a5ba4f0 Mon Sep 17 00:00:00 2001
From: Benjamin Loison <benjamin_loison@users.noreply.gitea.lemnoslife.com>
Date: Tue, 14 May 2024 02:12:54 +0200
Subject: [PATCH] Add titles

---
 datasets/raise/fft/fourier_to_image.py       |  1 +
 datasets/raise/fft/image_to_fourier.py       | 11 ++++++++++-
 datasets/raise/fft/remove_period_patterns.py |  3 ++-
 3 files changed, 13 insertions(+), 2 deletions(-)

diff --git a/datasets/raise/fft/fourier_to_image.py b/datasets/raise/fft/fourier_to_image.py
index 5fc411d..d1ba7d3 100644
--- a/datasets/raise/fft/fourier_to_image.py
+++ b/datasets/raise/fft/fourier_to_image.py
@@ -1,5 +1,6 @@
 from PIL import Image
 import numpy as np
+from scipy import fftpack
 
 fft = np.array(Image.open('fft.png').convert('L'))# / 255
 print(fft)
diff --git a/datasets/raise/fft/image_to_fourier.py b/datasets/raise/fft/image_to_fourier.py
index 30970b4..8bf2e1f 100644
--- a/datasets/raise/fft/image_to_fourier.py
+++ b/datasets/raise/fft/image_to_fourier.py
@@ -1,5 +1,6 @@
 from PIL import Image
 import numpy as np
+from scipy import fftpack
 
 image = np.array(Image.open('image_4.png').convert('L'))# / 255
 print(image)
@@ -8,8 +9,16 @@ print(image)
 # fftpack.fftshift
 fft1 = (fftpack.fft2(image))
 
+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)))
 plt.show()
-#plt.imsave('ifft1.png', fft1)
\ No newline at end of file
+#plt.imsave('ifft1.png', np.log10(abs(fft1)))
\ No newline at end of file
diff --git a/datasets/raise/fft/remove_period_patterns.py b/datasets/raise/fft/remove_period_patterns.py
index 71ba76e..6b86884 100644
--- a/datasets/raise/fft/remove_period_patterns.py
+++ b/datasets/raise/fft/remove_period_patterns.py
@@ -47,8 +47,9 @@ fixedImage = interpolate_replace_nans(fft1, kernel)
 
 figure, axes = plt.subplots(1, 2, sharex = True, sharey = True)
 
+axes[0].set_title('Original FFT')
 axes[0].imshow(np.log10(originalFft1))
+axes[1].set_title('FFT with significant lines attenuated')
 axes[1].imshow(np.log10(fixedImage))
-#plt.imshow(np.log10(fixedImage))
 plt.show()
 #plt.imsave('fft.png', np.log10(fixedImage))
\ No newline at end of file