Skip to content

Instantly share code, notes, and snippets.

@suraj-deshmukh
Forked from panzerama/hough_transform.py
Created January 15, 2020 17:58
Show Gist options
  • Save suraj-deshmukh/b1abfcf7e7c631c3af7a59c43c42e899 to your computer and use it in GitHub Desktop.
Save suraj-deshmukh/b1abfcf7e7c631c3af7a59c43c42e899 to your computer and use it in GitHub Desktop.
Solution for correcting text skew in image with scipy, scikit
#image processing resources
from skimage.io import imread, imshow
from skimage.filters import gaussian, threshold_otsu
from skimage.feature import canny
from skimage.transform import probabilistic_hough_line, rotate
#testing
import numpy as np
import os
#deskewing function
def deskew(filename):
image = imread(filename, as_grey=True)
#threshold to get rid of extraneous noise
thresh = threshold_otsu(image)
normalize = image > thresh
# gaussian blur
blur = gaussian(normalize, 3)
# canny edges in scikit-image
edges = canny(blur)
# hough lines
hough_lines = probabilistic_hough_line(edges)
# hough lines returns a list of points, in the form ((x1, y1), (x2, y2))
# representing line segments. the first step is to calculate the slopes of
# these lines from their paired point values
slopes = [(y2 - y1)/(x2 - x1) if (x2-x1) else 0 for (x1,y1), (x2, y2) in hough_lines]
# it just so happens that this slope is also y where y = tan(theta), the angle
# in a circle by which the line is offset
rad_angles = [np.arctan(x) for x in slopes]
# and we change to degrees for the rotation
deg_angles = [np.degrees(x) for x in rad_angles]
# which of these degree values is most common?
histo = np.histogram(deg_angles, bins=180)
# correcting for 'sideways' alignments
rotation_number = histo[1][np.argmax(histo[0])]
if rotation_number > 45:
rotation_number = -(90-rotation_number)
elif rotation_number < -45:
rotation_number = 90 - abs(rotation_number)
return rotation_number
# testing on image library
_path = '/path/to/images/'
for dirpath, dirnames, filenames in os.walk(_path):
for filename in filenames:
if filename.lower().endswith('.jpg') or filename.lower().endswith('.png'):
rotation_angle = deskew(_path + filename)
print('Rotation angle for ' + filename + ' is {0}'.format(rotation_angle) + '\n')
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment