matlab 函数 strel("line") 到 python

matlab function strel("line") to python

我想在 python

中使用 matlab 函数 strel("line")

我发现 python 像 scikit-learn / opencv / mahotas 这样的库

但是我找不到。

最后,我在pymorph 'seline'中找到了类似的函数,但它与matlab strel函数不同。


具体来说,我想使用(或实现)strel("line") 并旋转它。

喜欢 strel("line", 长度, 度数)


实例
a = strel("line",5,0)
b = strel("line",5,30)
c = strel("line",5,45)
输出是

像这样。

谁知道matlab的strel("line",length,degree)函数的算法或者python库等于strel("line",length,degree)的算法,请教我知道。谢谢你。

这类题可以随时查看项目Octave提供的代码。

使用this link你可以看到函数strel是如何在Octave中实现的。

以下代码完全摘自Octave的strel函数,对应案例strel('line'):

## Parameters (degrees and linelenght)
degrees = 30
linelen = 5


## Line length are always odd, to center strel at the middle of the line.
## We look it as a diameter of a circle with given slope
deg90 = mod (degrees, 90);
if (deg90 > 45)
   alpha = pi * (90 - deg90) / 180;
else
   alpha = pi * deg90 / 180;
endif
   ray = (linelen - 1)/2;

## We are interested only in the discrete rectangle which contains the diameter
## However we focus our attention to the bottom left quarter of the circle,
## because of the central symmetry.
c = round (ray * cos (alpha)) + 1;
r = round (ray * sin (alpha)) + 1;

## Line rasterization
line = false (r, c);
m = tan (alpha);
x = [1:c];
y = r - fix (m .* (x - 0.5));
indexes = sub2ind ([r c], y, x);
line(indexes) = true;

## We view the result as 9 blocks.
# Preparing blocks
linestrip = line(1, 1:c - 1);
linerest = line(2:r, 1:c - 1);
z = false (r - 1, c);

# Assemblying blocks
SE.nhood =  vertcat (
    horzcat (z, linerest(end:-1:1,end:-1:1)),
    horzcat (linestrip, true, linestrip(end:-1:1,end:-1:1)),
    horzcat (linerest, z(end:-1:1,end:-1:1))
    );

# Rotate/transpose/flip?
sect = fix (mod (degrees, 180) / 45);
switch (sect)
    case 1, SE.nhood = transpose (SE.nhood);
    case 2, SE.nhood = rot90 (SE.nhood, 1);
    case 3, SE.nhood = fliplr (SE.nhood);
    otherwise, # do nothing
endswitch

SE