Question

我尝试在频域中使用2D循环卷积进行时域乘法。实际上我知道它在一维情况下是如何工作的。像：

x = [1 2 3 4 5];

y = [4 5 6 7 8];

xy = 1/5*ifft( cconv(fft(x), fft(y), 5));

xy0 = x.*y;

xy和xy0都是一样的，这就是我想要的。但是，对于2D情况，cconv没有在matlab中定义，我也不知道如何在频域中使用卷积在相同大小的2个矩阵之间进行乘法。

假设我们有以下矩阵：

x = [3 5 4;
    7 6 1;
    -1 2 0];

y = [2 7 1;
    2 -3 2;
    5 6 9];

确定命令1/9 * ifft2（conv2（fft2（x），fft2（y），＆＃39;相同＆＃39;））不会给出与x相同的结果。* y

每个人都可以帮我解决这个问题吗？

Answer 1

您的卷积在您的命令中应为圆形卷积，与一维示例的cconv相同。参见https://www.mathworks.com/matlabcentral/answers/59333-convolution-in-frequency-domain-not-convolution-in-time-domain和Convolution of two fft function。

这是我必须使用Numpy和Scipy在Python中工作的示例。在IFFT之前，我不得不将卷积的结果在频域中偏移（+1 +1），以匹配乘法结果。我假设scipy pad会将结果从正确的位置移位，对空间域FFT的快速检查显示正确的FFT转换输出应该让您知道如何在matlab中或使用其他填充进行移位。

在scipy.signal中使用'wrap'填充等效于循环卷积。

import numpy as np
from scipy import signal

###################################################
# Check Multiplication in Spatial == Conv in Freq #
###################################################

np.random.seed(1234)
fr = np.random.normal(0.1,1.0,(3,3))
imgs = np.random.normal(0,1.0,(3,3))

spatialMult = fr * imgs

filtFFT = np.fft.fft2(fr)
imgsFFT = np.fft.fft2(imgs)

ac = signal.convolve2d(imgsFFT.real,filtFFT.real,'same','wrap')
ad = signal.convolve2d(imgsFFT.real,filtFFT.imag,'same','wrap')
bc = signal.convolve2d(imgsFFT.imag,filtFFT.real,'same','wrap')
bd = signal.convolve2d(imgsFFT.imag,filtFFT.imag,'same','wrap')

r_conv = (ac - bd)
i_conv = (ad + bc)

res_conv = r_conv + 1j*i_conv
res_conv = res_conv / 9
# Positions are incorrect for some reason
# Need a +1 +1 shift to align
res_conv = np.roll(np.roll(res_conv,1,0),-2,1)
res_ifft = np.fft.ifft2(res_conv)

spat_fft = np.fft.fft2(spatialMult)
print(spat_fft)
print(res_conv)
print()
print(res_ifft.real)
print(spatialMult)

如果您希望立即获得正确的输出，则可以使用numpy自己进行填充：

ir = imgsFFT.real
im = imgsFFT.imag
f_r = filtFFT.real
f_i = filtFFT.imag

ir = np.pad(ir,((2,0),(2,0)),'wrap')
im = np.pad(im,((2,0),(2,0)),'wrap')

ac = signal.convolve2d(ir,f_r,'valid')
ad = signal.convolve2d(ir,f_i,'valid')
bc = signal.convolve2d(im,f_r,'valid')
bd = signal.convolve2d(im,f_i,'valid')

r_conv = (ac - bd)
i_conv = (ad + bc)

res_conv = r_conv + 1j*i_conv
res_conv = res_conv / n
res_ifft = np.fft.ifft2(res_conv)

spat_fft = np.fft.fft2(spatialMult)
print(spat_fft)
print(res_conv)
print()
print(res_ifft.real)
print(spatialMult)

在频域中使用2D循环卷积在时域中的乘法

1 个答案: