如何在matplotlib,scipy等中连接不连续曲线

时间:2012-06-17 22:57:07

标签: python numpy matplotlib scipy

我有一个小问题,当我拍摄图像的轮廓时,我得到了这个数字:

正如你所看到的,我可以提取轮廓,但是一旦我提取了路径,它就会留下这些奇怪的割线,它将穿过图像,因为曲线上有2个不连续的区域。我想知道是否有一种方法可以断开不连续的线路,或者是我的路径提取代码是错误的

import matplotlib.pyplot as plt
import numpy as np

def contourPath(img, width, height): 
    x         = np.arange(0,width)
    y         = np.arange(height,0,-1)
    X, Y      = np.meshgrid(x,y)
    plot      = plt.contour(X,Y,img, [0])
    pathList  = plot.collections[0].get_paths()
    x, y      = [], []
    for i in range(0, len(pathList)):
        iterPath = pathList[i].iter_segments()
        for point in iterPath:
            pt  = np.rint(point[0])
            x.append(pt[0])
            y.append(pt[1])
    X         =  np.hstack(x)
    Y         =  np.hstack(y)
    return np.dstack((X,Y))[0]

感谢您的时间

for user545424 我想在这里。 Matplotlib轮廓功能正常工作,因为图像上有两个不连续的点,导致这个小事件发生。

我知道这些割线是由scypi引起的,但它引发了关于库如何与轮廓点相互作用的另一个问题

哦,我相信可以通过查找路径并对其进行插值来掩盖问题。但是,我喜欢避免重新修改路径,因为在我的计算机上旅行Salemans问题并不好。

你有什么建议吗?

enter image description here

1 个答案:

答案 0 :(得分:3)

我想我正在回答我自己的问题。那些奇怪的割线是由scipy汇总列表的末端引起的。默认行为,我相信scipy将连接2个连续点,无论位置如何。顶点工作正常,因为它严格地采用了这个图像的轮廓并且它包围了曲线。割线连接到顶部半圆的中间的原因。这是最后一条路径结束的位置。

我想转换到极地并重新修改路径很简单,但不是很完美,但是很好。

我想知道其他人是否有更好的解决方案

   def cart2polar(x,y, origin=None, size=np.array([200,200])):
      ny, nx= size[0], size[1]
      print size.shape
      if origin is None:
          origin_x, origin_y = nx//2, ny//2
      else:
          origin_x, origin_y = origin[0], origin[1]
      x -= origin_x
      y -= origin_y
      r = np.sqrt(x**2 + y**2)
      theta = np.arctan2(y, x)
      return r, theta

    def main():
      index = np.argsort(theta)
      r, theta = r[index[:]], theta[index[:]]
      f = interpolate.interp1d(theta,r)
      theta = np.linspace(round(theta.min()+.00005,),theta.max(), 200)
      r = f(theta)
      x,y = polar2cart(r, theta)

    def polar2cart(r, theta, origin=None, size=np.array([200,200]) ):
      ny, nx= size[0], size[1]
      if origin is None:
          origin_x, origin_y = nx//2, ny//2
      else:
          origin_x, origin_y = origin[0], origin[1]
      x += origin_x
      y += origin_y

      return x, y