ae_python_fft.py

import numpy as np
import cv2


def ParamsSetup():
    params = (
        ('FFT Division', 'float', {'value': 12.0, 'minvalue': 1.0, 'maxvalue': 1000.0}),
    )
    return {'params': params, 'mode': 'RGB'}


def SmartRender(in_data, params, input_):
    gray = cv2.cvtColor(input_, cv2.COLOR_RGB2GRAY)
    fft = np.fft.fft2(gray)
    fft = np.log(np.fft.fftshift(np.abs(fft)) + 1.0)
    fft_division = max(1.0, params['FFT Division'])
    fft /= fft_division
    return fft