如何使用python和opencv检测图像中的水平线并获得其y坐标？

Question

我正在使用find contours方法,然后使用fitline函数逼近一条线.以下是代码:

img = cv2.imread('lines.jpg')
imgray = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
ret,dst = cv2.threshold(imgray,127,255,0)
im2,cnts, hierarchy =cv2.findContours(dst,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)
rows,cols = img.shape[:2]
[vx,vy,x,y] = cv2.fitLine(cnts[0], cv2.DIST_L2,0,0.01,0.01)
lefty = int((-x*vy/vx) + y)
righty = int(((cols-x)*vy/vx)+y)
cv2.line(img,(cols-1,righty),(0,lefty),(0,255,0),2)
print img.shape[:2]
cv2.imshow('image1',img)
cv2.waitKey(0)
cv2.destroyAllWindows()

以下是我作为输出获得的图像

我期待它检测图像中的每个黑色条纹,而它只检测从底部的第一行

Answer 1

这个修改过的代码是您可能一直在寻找的.我已经离开了你的大部分行,而且非常冗长,但这可能会帮助你理解它而无需进一步解释.

我认为你陷入了两个不同的问题:

• 代码中的cnts [0]仅指第一个轮廓.我已经介绍了一个参数contnumber,你可能需要循环.我当前的代码仅对轮廓号4执行

• 您可能没有将线条检测为前景,但是白色区域,这就是我否定图像的原因(thresh =(255-thresh))

  import numpy as np
  import cv2
  
  im = cv2.imread('lines.jpg')
  rows,cols = im.shape[:2]
  imgray = cv2.cvtColor(im,cv2.COLOR_BGR2GRAY)
  ret,thresh = cv2.threshold(imgray,125,255,0)
  thresh = (255-thresh)
  thresh2=thresh.copy()
  im2, contours, hierarchy = cv2.findContours(thresh,cv2.RETR_TREE,cv2.CHAIN_APPROX_SIMPLE)
  
  cv2.imshow('image1',im)
  cv2.imshow('image3',thresh2)
  #cv2.drawContours(im, contours, -1, (0,255,0), 3) #draw all contours
  contnumber=4
  cv2.drawContours(im, contours, contnumber, (0,255,0), 3) #draw only contour contnumber
  cv2.imshow('contours', im)
  
  [vx,vy,x,y] = cv2.fitLine(contours[contnumber], cv2.DIST_L2,0,0.01,0.01)
  lefty = int((-x*vy/vx) + y)
  righty = int(((cols-x)*vy/vx)+y)
  cv2.line(im,(cols-1,righty),(0,lefty),(0,255,255),2)
  
  cv2.imshow('result', im)
  
  cv2.waitKey(0)
  cv2.destroyAllWindows()
  

希望有所帮助.