You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.2 KiB
46 lines
1.2 KiB
# -*- coding: utf-8 -*-
|
|
import cv2
|
|
import numpy as np
|
|
|
|
def LineCheck(img):
|
|
|
|
img = cv2.GaussianBlur(img, (3, 3), 0)
|
|
edges = cv2.Canny(img, 50, 150, apertureSize=3)
|
|
|
|
|
|
lines = cv2.HoughLines(edges,1,np.pi/2,118)
|
|
|
|
######### end ##########
|
|
|
|
result = img.copy()
|
|
for i_line in lines:
|
|
for line in i_line:
|
|
rho = line[0]
|
|
theta = line[1]
|
|
if (theta < (np.pi / 4.)) or (theta > (3. * np.pi / 4.0)): # 垂直直线
|
|
pt1 = (int(rho / np.cos(theta)), 0)
|
|
pt2 = (int((rho - result.shape[0] * np.sin(theta)) / np.cos(theta)), result.shape[0])
|
|
cv2.line(result, pt1, pt2, (0, 0, 255))
|
|
else:
|
|
pt1 = (0, int(rho / np.sin(theta)))
|
|
pt2 = (result.shape[1], int((rho - result.shape[1] * np.cos(theta)) / np.sin(theta)))
|
|
cv2.line(result, pt1, pt2, (0, 0, 255), 1)
|
|
|
|
|
|
minLineLength = 200
|
|
maxLineGap = 15
|
|
lines=None
|
|
|
|
|
|
linesP=cv2.HoughLinesP(edges,1,np.pi/180,80,minLineLength,maxLineGap)
|
|
|
|
######### end ##########
|
|
|
|
|
|
result_P = img.copy()
|
|
for i_P in linesP:
|
|
for x1, y1, x2, y2 in i_P:
|
|
cv2.line(result_P, (x1, y1), (x2, y2), (0, 255, 0), 3)
|
|
|
|
|
|
return result_P |