OpenCV实现文档倾斜矫正的两种方法

Ruby
Ruby
发布时间:2026-10-15| 4人浏览过
原创

本文介绍利用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)

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持本站。

免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。

Python 相关产品和应用

全部下载
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架,支持类型提示和自动接口文档。

Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

Django

Django

6.1.2 · Windows

成熟的 Python Web 框架,适合构建内容站点和业务系统。

Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。