当我尝试在matlab中执行非最大抑制时，图像是一条线

Question

所以，我在matlab中编写了这段代码，应该执行非最大抑制。本质上，它应该将给定点与其相邻点进行比较，如果它比所有相邻点都高，则将该点设为1，否则设为0。

当我运行代码时，我拥有的图像是一行。错误可能在哪里

<function newMagnitudeImage = NonMaximalSuppression(magnitude,orientation)
[m,n]=size('Brainweb');
% Discretization of directions
orientationdis= zeros(m,n);
for i = 1  : m
    for j = 1 : n
        if ((orientation(i, j) > 0 ) && (orientation(i, j) < (pi/8)) || (orientation(i, j) > (7*pi/8)) && (orientation(i, j) < (-7*pi/8)))
           orientationdis(i, j) = 0;
        end
        if ((orientation(i, j) > (pi/8)) && (orientation(i, j) < (3*pi/8)) || (orientation(i, j) < (-5*pi/8)) && (orientation(i, j) > (-7*pi/8)))
            orientationdis(i, j) = pi/4;
        end
        if ((orientation(i, j) > (3*pi/8)) && (orientation(i, j) < (5*pi/8)) || (orientation(i, j) < (-3*pi/8)) && (orientation(i, j) > (5*pi/8)))
            orientationdis(i, j) = pi/2;
        end
        if ((orientation(i, j) > (5*pi/8) && (orientation(i, j) <= (7*pi/8)) || (orientation(i, j) < (-pi/8) && (orientation(i, j) > (-3*pi/8)))))
            orientationdis(i, j) = 3*pi/4;
        end
    end
end
newMagnitudeImage = zeros(m, n);
for i = 2  : m-1
    for j = 2 : n-1
        if (orientationdis(i, j) == 0)
            if (magnitude(i, j) > magnitude(i, j - 1) && magnitude(i, j) > magnitude(i, j + 1))
                newMagnitudeImage(i, j) = magnitude(i, j);
            else
               newMagnitudeImage(i, j) = 0;
            end
        end
        if (orientationdis(i, j) == 45)
            if (magnitude(i, j) > magnitude(i + 1, j - 1) && magnitude(i, j) > magnitude(i - 1, j + 1))
                newMagnitudeImage(i, j) = magnitude(i, j);
            else
               newMagnitudeImage(i, j) = 0;
            end
        end
        if (orientationdis(i, j) == 90)
            if (magnitude(i, j) > magnitude(i - 1, j) && magnitude(i, j) > magnitude(i + 1, j))
                newMagnitudeImage(i, j) = magnitude(i, j);
            else
                newMagnitudeImage(i, j) = 0;
            end
        end
        if (orientationdis(i, j) == 135)
            if (magnitude(i, j) > magnitude(i - 1, j - 1) && magnitude(i, j) > magnitude(i + 1, j + 1))
                newMagnitudeImage(i, j) = magnitude(i, j);
            else
               newMagnitudeImage(i, j) = 0;
            end
        end
    end
end

Answer 1

我希望我能理解，但是你可以用nlfilter应用滑动邻域操作。

例如:

I = randi([1,10],10,10);
fun = @(x) max(x(:));
I2 = nlfilter(I,[3 3],fun); %calculate the maximum of the neighborhood.
ind = I==I2; %is each element a local maximum ?
%suppression if the value is not the maximum of the neighborhood.
I(~ind) = NaN;

nlfilter需要图像处理工具箱

Answer 2

在代码的上半部分，你用弧度在0到3*pi/4之间离散，但在下半部分，你用度来检查。您应该更改这两个，使它们匹配(即将==45更改为== pi/4，等等)

看起来代码中可能有更多的问题-为什么你只在0和135度之间离散?

编辑:相同代码的更短版本[需要图像处理工具箱]:

orientationdis = mod(round(orientation/(pi/4)),4)*pi/4 %map orientation to correct value of from 0 to pi, rounded to the nearest pi/4
newMagnitudeImage = (imdilate(orientation,[0 1 0; 1 1 1; 0 1 0])==orientation).*magnitude; %Find the maximum value in each neighborhood, compare to the original values, set non-maximum values to 0 and maximum values to original value from magnitude

如果您没有在代码的其他地方使用orientationdis，您只需要第二行