子矩阵（和子向量）上的操作可以向量化吗？

Question

我目前正在研究 Octave 的边缘检测器。来自 Java 和 Python 等其他编程语言，我习惯于在 for 循环中迭代，而不是对整个矩阵执行操作。现在在八度音程中，这会导致严重的性能下降，而且我在弄清楚如何矢量化我的代码时遇到了一些困难。我有以下两段代码：

1)

function zc = ZeroCrossings(img, T=0.9257)
  zc = zeros(size(img));

  # Iterate over central positions of all 3x3 submatrices
  for y = 2:rows(img) - 1
    for x = 2:columns(img) - 1
      ndiff = 0;

      # Check all necessary pairs of elements of the submatrix (W/E, N/S, NW/SE, NE/SW)
      for d = [1, 0; 0, 1; 1, 1; 1, -1]'
        p1 = img(y-d(2), x-d(1));
        p2 = img(y+d(2), x+d(1));
        if sign(p1) != sign(p2) && abs(p1 - p2) >= T
          ndiff++;
        end
      end

      # If at least two pairs fit the requirements, these coordinates are a zero crossing
      if ndiff >= 2
        zc(y, x) = 1;
      end
    end
  end
end

2)

function g = LinkGaps(img, k=5)
  g = zeros(size(img));

  for i = 1:rows(img)
    g(i, :) = link(img(i, :), k);
  end
end

function row = link(row, k)
  # Find first 1
  i = 1;
  while i <= length(row) && row(i) == 0
    i++;
  end

  # Iterate over gaps
  while true
    # Determine gap start
    while i <= length(row) && row(i) == 1
      i++;
    end
    start = i;

    # Determine gap stop
    while i <= length(row) && row(i) == 0
      i++;
    end

    # If stop wasn't reached, exit loop
    if i > length(row)
      break
    end

    # If gap is short enough, fill it with 1s
    if i - start <= k
      row(start:i-1) = 1;
    end
  end
end

这两个函数都遍历子矩阵（或第二种情况下的行和子行），尤其是第一个似乎使我的程序变慢了很多。

此函数采用像素矩阵 (img) 和 returns 二进制 (0/1) 矩阵，其中 1s 为零交叉点（对应 3x3 邻域的像素符合某些要求）被发现。

外面的 2 个 for 循环似乎应该可以以某种方式进行矢量化。我可以将主体放入它自己的函数中（将必要的子矩阵作为参数）但我不知道如何在所有子矩阵上调用这个函数，将它们相应的（中心）位置设置为返回值。

如果内部 for 循环也可以矢量化，则加分。
此函数从前一个输出中获取二进制矩阵，并填充其行中的空白（即将它们设置为 1）。间隙定义为长度 <= k 的一系列 0，两侧以 1 为界。

现在我确定至少外层循环（LinkGaps 中的循环）是可向量化的。但是，link 中的 while 循环再次作用于子向量，而不是单个元素，所以我不确定我将如何对其进行向量化。

Answer 1

不是一个完整的解决方案，但这里有一个想法，您可以如何在没有任何循环的情况下完成第一个：

% W/E
I1 = I(2:end-1,1:end-2);
I2 = I(2:end-1,3:end  );
C = (I1 .* I2 < 0) .* (abs(I1 - I2)>=T);

% N/S
I1 = I(1:end-2,2:end-1);
I2 = I(3:end,  2:end-1);
C = C + (I1 .* I2 < 0) .* (abs(I1 - I2)>=T);

% proceed similarly with NW/SE and NE/SW
% ...

% zero-crossings where count is at least 2
ZC = C>=2;

思路：形成两个适当偏移的子图像，检查符号差异（乘积负片）并对差异设置阈值。两个测试 return 一个逻辑 (0/1) 矩阵，element-wise 产品执行逻辑，结果是一个 0/1 矩阵，其中两个测试都成功了 1。可以添加这些矩阵以跟踪计数 (ndiff)。

子矩阵（和子向量）上的操作可以向量化吗？

Can operations on submatrices (and subvectors) be vectorized?

matlab

vectorization

octave