0

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

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

<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
4

2 回答 2

0

我希望我能理解，但您可以使用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需要图像处理工具箱

于 2016-10-17T21:12:46.667 回答
0

在代码的上半部分，您在 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，则只需要第二行。

于 2016-10-17T18:16:01.213 回答