OpenCV实现文档倾斜矫正的两种方法
本文介绍利用OpenCV实现文档倾斜矫正的两种核心方法。第一种基于霍夫变换检测平行直线并计算平均角度;第二种通过形态学操作提取轮廓,利用最小包围矩形获取倾斜角。文章提供了完整的Python代码示例及图像旋转的具体实现参数说明,帮助开发者快速完成文档图像的自动校正。
本文分享使用OpenCV进行文档图像矫正的具体实现代码,供读者参考学习。
原始文档图像

矫正后的文档图像

实现思路
文档矫正的核心在于获取倾斜角度,随后通过仿射变换进行旋转校正。获取倾斜角度的方法主要有两种,下文将分别详细介绍。
第一种方法利用霍夫变换。由于文档内容通常由平行线条组成,首先使用霍夫变换检测图像中的直线,计算所有检测到的直线的平均倾斜角度,将其视为文档的整体倾斜角,最后执行仿射变换完成校正。
import cv2
import numpy as np
def imshow(img):
cv2.imshow("img",img)
cv2.waitKey(0)
cv2.destroyAllWindows()
img = cv2.imread("2.png",1)
img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
img2 = img.copy()
img_canny = cv2.Canny(img,75,200)
img_line = cv2.HoughLines(img_canny,1,np.pi/180,280)
average = 0
for line in img_line:
for rho,theta in line:
average = average + theta
average = average / len(img_line)
angel = average/np.pi * 180 - 90
def rotateImg(img,angel):
rows, cols = img.shape
M = cv2.getRotationMatrix2D(((cols - 1) / 2.0, (rows - 1) / 2.0), angel, 1) # 旋转中心x,旋转中心y,旋转角度,缩放因子
img = cv2.warpAffine(img, M, (cols, rows),borderValue = (255,255,255)) #在内存里完成了旋转
imshow(img)
rotateImg(img2,angle)第二种方法是求文档内容的最小包围矩形。首先检测图像轮廓,利用形态学操作生成掩膜(mask),再次检测轮廓并计算最小包围矩形。该矩形会返回一个倾斜角度(单位为度,而霍夫变换返回的是弧度制)。可以通过筛选轮廓,选取面积最大的轮廓所对应的倾斜角度作为文档角度,进而进行仿射变换。
import cv2
import numpy as np
def imshow(img):
cv2.imshow("img",img)
cv2.waitKey(0)
cv2.destroyAllWindows()
img = cv2.imread("2.png",1)
img = cv2.cvtColor(img,cv2.COLOR_BGR2GRAY)
img2 = img.copy()
imgth = cv2.threshold(img,0,255,cv2.THRESH_OTSU|cv2.THRESH_BINARY_INV)[1]
kernel = np.ones((17,17))
img_open = cv2.morphologyEx(imgth,cv2.MORPH_CLOSE,kernel,10)
# imshow(img_open)
cons = cv2.findContours(img_open,cv2.RETR_EXTERNAL,cv2.CHAIN_APPROX_NONE)[0]
areas = []
for con in cons:
area = cv2.contourArea(con)
areas.append(area)
index = areas.index(max(areas))
# 度数形式,霍夫变换的返回值是弧度制
rect = cv2.minAreaRect(cons[index])
angle = rect[2]
mat = cv2.getRotationMatrix2D((img.shape[1]/2,img.shape[0]/2),angle,1)
img_fin = cv2.warpAffine(img,mat,(img.shape[1],img.shape[0]),borderValue = (255,255,255))
imshow(img_fin)补充知识
利用仿射变换实现图像旋转指定度数
具体实现代码如下: mat = cv2.getRotationMatrix2D(center,angle,c) (center: 旋转中心,angle:旋转角度,c:缩放大小) img_final = cv2.warpAffine(img,mat, (img.shape[1],img.shape[0]), borderValue = (255,255,255)) (borderValue为可选参数,填充色,默认为黑色)
此外,若需旋转90度、180度或270度,可以使用transpose和flip函数实现。
旋转90度(顺时针) img = cv2.transpose(img) res_img = cv2.flip(img,1)
旋转180度 img = cv2.flip(img,0) img = cv2.flip(img,1)
以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持本站。
免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。









