十五、霍夫直线检测
一、自定義
import cv2 import numpy as np from matplotlib import pyplot as pltdef line_detection(image):gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)edges = cv2.Canny(gray,50,150,apertureSize=3)lines = cv2.HoughLines(edges,1,np.pi/180,200)for line in lines:rho,theta = line[0]a = np.cos(theta)b = np.sin(theta)x0 = a*rhoy0 = b*rhox1 = int(x0+1000*(-b))y1 = int(y0+1000*(a))x2 = int(x0-1000*(-b))y2 = int(y0-1000*(a))cv2.line(image,(x1,y1),(x2,y2),(0,0,255),2)#將直線畫到image上,直線的兩點分別為(x1,y1),(x2,y2),直線的顏色為(0,0,255),直線的寬度為2cv2.imshow("image_lines",image)src = cv2.imread(r"G:\Juptyer_workspace\study\opencv\opencv3\qqq.png") cv2.imshow("image",src) cv2.namedWindow("image",cv2.WINDOW_AUTOSIZE) line_detection(src) cv2.waitKey(0) cv2.destroyAllWindows()效果圖如下:
二、調用API接口
import cv2 import numpy as np from matplotlib import pyplot as pltdef line_detect(image):gray = cv2.cvtColor(image,cv2.COLOR_BGR2GRAY)edges = cv2.Canny(gray,50,150,apertureSize=3)lines = cv2.HoughLinesP(edges,1,np.pi/180,100,minLineLength=50,maxLineGap=10)for line in lines:x1,y1,x2,y2 = line[0]cv2.line(image,(x1,y1),(x2,y2),(0,0,255),2)#將直線畫到image上,直線的兩點分別為(x1,y1),(x2,y2),直線的顏色為(0,0,255),直線的寬度為2cv2.imshow("image_lines",image)src = cv2.imread(r"G:\Juptyer_workspace\study\opencv\opencv3\qqq.png") cv2.imshow("image",src) cv2.namedWindow("image",cv2.WINDOW_AUTOSIZE) line_detect(src) cv2.waitKey(0) cv2.destroyAllWindows()效果圖如下:
總結
- 上一篇: 空调冷媒多少钱啊?
- 下一篇: 十八、对已经找到轮廓的图像进行测量