二维图像(数组)的fftshift

功能

自行编写二维图像的fftshift，并进行测试

环境

Matlab

fftshift是啥

将零频分量移到频谱中心
Y = fftshift(X) 通过将零频分量移动到数组中心，重新排列傅里叶变换 X。
如果 X 是向量，则 fftshift 会将 X 的左右两半部分进行交换。
如果 X 是矩阵，则 fftshift 会将 X 的第一象限与第三象限交换，将第二象限与第四象限交换。
如果 X 是多维数组，则 fftshift 会沿每个维度交换 X 的半空间。

fftshift的效果

左边为fftshift前的效果，右边为fftshift后的效果（32*32）

定义8*8的二维数组，依次填充1:64

变换前后的效果如图所示

找规则

对变换前后的结果作差

可以发现，二维数组分为四个部分

地址变换关系可以总结为如上公式

测试

经过测试，自编的fftshift在效果上与fftshift完全相同，但是只能适用于偶数阶方阵，但是对于奇数阶方阵或者矩阵而言，也有其固有的关系，可以自行总结。

Code

以下代码不能直接运行，仅供参考。建议修改后分块运行，有问题可以评论区交流。

%% 生成地址方阵
m = 8;
n = 8;
addr = 1:m*n;
a_shrink = reshape(addr, m, n);
a_shrink = a_shrink';
disp("---");
disp(a_shrink);
figure; imshow(a_shrink, []);
%% 生成数据方阵
m = 8;
n = 8;
data = 1:n;
data = repmat(data', m, 1);
data_shrink = reshape(data, m, n);
data_shrink = data_shrink';
disp("---");
disp(data_shrink);
figure; imshow(data_shrink, []);
%% fftshift
a_f = fftshift(a_shrink);
disp("---");
disp(a_f);
figure; 
subplot(1,2,1); imshow(a_shrink, []);
subplot(1,2,2); imshow(a_f, []);
disp("---");
disp(a_f-a_shrink);

%% myfftshift
% +-(n*m/2 +-m/2)
a_addr_s = a_shrink;

a_addr_f(1:m/2, 1:n/2)      = a_addr_s(1:m/2, 1:n/2)     + (n*m/2 +m/2);
a_addr_f(1:m/2, n/2+1:n)    = a_addr_s(1:m/2, n/2+1:n)   + (n*m/2 -m/2);
a_addr_f(m/2+1:m, 1:n/2)    = a_addr_s(m/2+1:m, 1:n/2)   - (n*m/2 -m/2);
a_addr_f(m/2+1:m, n/2+1:n)  = a_addr_s(m/2+1:m, n/2+1:n) - (n*m/2 +m/2);

% disp(a_addr_f-a_f);
%% test myfftshift
load('I1.mat');
I1_expand = reshape(I1', 1, []); % 按行展开为行向量
I2_expand = zeros(1, m*n);
for i=1:m*n
   I2_expand(i) = I1_expand(a_addr_f(i));
end
I2 = reshape(I2_expand, m, n);
I2 = I2';

figure; 
subplot(1,2,1); imagesc(abs(I1));        axis square; axis on; colormap(jet(255)); colorbar; title('before');
subplot(1,2,2); imagesc(abs(I2));        axis square; axis on; colormap(jet(255)); colorbar; title('after');

FPGA实现

将变换后的地址保存到ROM中，在变换时，从原位置读取数据，按ROM中的地址保存数据。

---

看到这儿了，点个赞吧！