将2D FFT应用于图像

Question

我一直在尝试使用TypeScript实现FFT，我想将其应用于图像数据（这是一个128x128的灰度2D数组）。我正在考虑以下步骤，但是我不确定自己是否朝着正确的方向前进。

图像数据：

[[1 2 3],
 [4 5 6],
 [7 8 9]]

1. 将3x3图像数据数组平铺为1x9（例如[1 2 3 4 5 6 7 8 9]）
2. 应用FFT并计算频谱（对于x轴）
3. 将频谱阵列重塑为3x3
4. 转置频谱数组
5. 将3x3频谱阵列展平为1x9（例如[1 4 7 2 5 8 3 6 9]）
6. 应用FFT并计算频谱（针对y轴）
7. 与3相同
8. 与4相同
9. 计算功率谱并绘制图

我实际上得到了一个非常熟悉的结果，但是不知何故我没有信心。

运行中的脚本：https://tatsuky.com/playground/fft-ts/