如何在OpenCV中找到T形的角度

时间:2016-09-24 23:15:18

标签: python opencv

我正在使用OpenCV 2.4进行一些跟踪,我可以获得我想要的形状轮廓,这是一个T.

输入图片: enter image description here

我可以使用cv2.minAreaRect(my_t_contour)并获得该矩形的角度,但这只能给我0-180度。但这是一个T形,所以我希望能够告诉0-360。我在考虑:

  1. 将轮廓分成两个部分
  2. 通过rects(使用skeletonize> HoughLinesP)获取一行
  3. 确定哪条线,确定它们的渐变(使用我从HoughLinesP得到的坐标),然后确定T的方向。
  4. 但是我被困在1号,我怎么能将轮廓分成两个形状?

    方法1 :绘制轮廓中心和轮廓的minAreaRect

    dst = cv2.cvtColor(r_target, cv2.COLOR_BGR2GRAY)
    dst = cv2.GaussianBlur(dst, (11, 11), 0)
    ret,dst = cv2.threshold(dst,110,255,cv2.THRESH_BINARY_INV)
    cnts = cv2.findContours(dst, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)
    for c in cnts:
        # get minAreaRect around contour and draw its center in red
        rect = cv2.minAreaRect(c)
        cv2.circle(r_target, (int(rect[0][0]), int(rect[0][1])), 7, (0, 0, 255), -1)
    
        # get moments of contour to get center and draw it in white
        M = cv2.moments(c)
        cX = int(M["m10"] / M["m00"])
        cY = int(M["m01"] / M["m00"])
        cv2.circle(r_target, (cX, cY), 7, (255, 255, 255), -1)
    

    enter image description here

    下一步可能会计算中心之间的简单渐变来确定角度。

    方法2 :使用HoughLinesP对图像进行骨架化并获取线条。

    dst = cv2.cvtColor(r_target, cv2.COLOR_BGR2GRAY)
    dst = cv2.GaussianBlur(dst, (11, 11), 0)
    ret,dst = cv2.threshold(dst,110,255,cv2.THRESH_BINARY)
    dst = 1 - dst / 255
    dst = skimage.morphology.skeletonize(dst).astype(np.uint8)
    rho = 1
    theta = np.pi / 180
    threshold = 1
    minLineLength = 30
    maxLineGap = 15
    lines = cv2.HoughLinesP(dst, rho, theta, threshold, minLineLength=minLineLength, maxLineGap=maxLineGap)
    for line in lines[0]:
        cv2.line(r_target, (line[0], line[1]), (line[2], line[3]), (0, 255, 0), 1, 8)
    

    但这些线条并不是很好。这就是骨架的样子:

    enter image description here

    我还在尝试使用HoughLinesP吗?

    我还在尝试变量吗?

1 个答案:

答案 0 :(得分:1)

作为变体,您可以使用PCA,找到第一个组件方向,并将其用作双角。您可以在此处查看示例:http://docs.opencv.org/trunk/d1/dee/tutorial_introduction_to_pca.html