Python lane angle detection

from skimage.transform import (hough_line, hough_line_peaks)
import numpy as np
import cv2

image = cv2.imread('2.png')

# Compute arithmetic mean
image = np.mean(image, axis=2)

# Perform Hough Transformation to detect lines
hspace, angles, distances = hough_line(image)

# Find angle
angle=[]
for _, a , distances in zip(*hough_line_peaks(hspace, angles, distances)):
    angle.append(a)

# Obtain angle for each line
angles = [a*180/np.pi for a in angle]

# Compute difference between the two lines
angle_difference = np.max(angles) - np.min(angles)
print(angle_difference)

Source: stackoverflow.com

Similar codes

2 votes285 views1 answer
Reverse order np array
Python
7 votes281 views2 answers
What is the django orm
Python Django
0 vote275 views1 answer
Calculate pearson's r | calculate graph correlation
Python
0 vote305 views1 answer
Pyqt5 line edit password input
Python
0 vote268 views1 answer
How to count number of records in json
Python
124 votes267 views6 answers
Pandas rename column
Python
3 votes235 views1 answer
Endswith python
Python
0 vote249 views1 answer
Char count in python
Python
7 votes312 views2 answers
Create the dataframe column based on condition
Python
1 vote235 views1 answer
Check if pandas dataframe contains inf
Python