Гауссовый фильтр Notch в питоне ⇐ Python

[Anonymous]

Я пытаюсь спроектировать гауссовый фильтр Notch в питоне, чтобы удалить периодический шум. Я пытался реализовать следующую формулу: < /p>

Гауссовый фильтр notch < /p>

И вот код: < /p>

и вот код: < /p>

и вот код: < /p>

и вот код: < /p>

и вот код: < /p>

и вот код: < /p>

и вот код: < /p>

import numpy as np

def gaussian_bandpass_filter(image):
image_array = np.array(image)
#Fourier Transform
fourier_transform = np.fft.fftshift(np.fft.fft2(image_array))

#Size of Image
m = np.shape(fourier_transform)[0]
n = np.shape(fourier_transform)[1]

u = np.arange(m)
v = np.arange(n)

# Find the center
u0 = int(m/2)
v0 = int(n/2)

# Bandwidth
D0 = 10

gaussian_filter = np.zeros(np.shape(fourier_transform))

for x in u:
for y in v:
D1 = math.sqrt((x-m/2-u0)**2 + (y-n/2-v0)**2)
D2 = math.sqrt((x-m/2+u0)**2 + (y-n/2+v0)**2)
gaussian_filter[x][y] = 1 - math.exp(-0.5 * D1*D2/(D0**2))

#Apply the filter
fourier_transform = fourier_transform + gaussian_filter
image_array = np.fft.ifft2(np.fft.ifftshift(fourier_transform))

return image_array
< /code>

Предполагается, что эта функция применяет гауссовый фильтр на изображение и возвращает фильтрованное изображение, но, похоже, не работает. Я не знаю, где я ошибся с этим (может быть, я не понял формулу правильно?), Поэтому, если кто -то мог бы мне помочь, я бы очень признателен. приемлемо.
Но моя функция возвращает это. (Я использую pil.image.fromarray для преобразования массива в изображение)

Подробнее здесь: https://stackoverflow.com/questions/599 ... -in-python