我需要对两个一维信号进行卷积,其中一个平均有500个点(这个是汉宁窗函数),另一个是125000.每次运行时,我需要应用三次卷积运算。我已经有一个基于scipy文档的实现。你可以在这里看到代码,如果你想(Delphi代码提前):快速卷积算法
function Convolve(const signal_1, signal_2 : ExtArray) : ExtArray;
var
capital_k : Integer;
capital_m : Integer;
smallest : Integer;
y : ExtArray;
n : Integer;
k : Integer;
lower, upper : Integer;
begin
capital_k := Length(signal_1) + Length(signal_2) - 1;
capital_m := Math.Max(Length(signal_1), Length(signal_2));
smallest := Math.Min(Length(signal_1), Length(signal_2));
SetLength(y, capital_k);
for n := 0 to Length(y) - 1 do begin
y[n] := 0;
lower := Math.Max(n - capital_m, 0);
upper := Math.Min(n, capital_k);
for k := lower to upper do begin
if (k >= Length(signal_1)) or (n - k >= Length(signal_2)) then
Continue;
y[n] := y[n] + signal_1[k] * signal_2[n - k];
end;
end;
Result := Slice(y,
Floor(smallest/2) - 1,
Floor(smallest/2) - 1 + capital_m);
end;
问题是,这个实现太慢了。整个过程大约需要五分钟。我想知道如果我能找到更快的计算方法。
什么是零填充,需要?它确实使信号成为二的幂次? – Sambatyon 2011-05-18 11:54:37
@Sambatyon:你需要零填充来使内核的大小与信号的大小相同(除非你使用overlap-add或overlap-save方法),你也需要它来将循环卷积转换为线性卷积。 – 2011-05-18 12:14:17
+1,FFT-multiply-IFFT绝对是您的选择。在你的情况下,把两个信号都加到(至少)125500点以避免循环包裹效应。根据您使用/查找哪种FFT实现,您可能需要以2的乘方来获得速度。 – mtrw 2011-05-18 20:51:50