当前位置：首页 > news >正文

计算机视觉（opencv）实战十八——图像透视转换

news 2025/9/14 6:07:15

图像透视变换详解与实战

在图像处理中，透视变换（Perspective Transform） 是一种常见的几何变换，用来将图像中某个四边形区域拉伸或压缩，映射到一个矩形区域。常见应用场景包括：

纠正拍照时的倾斜（如手机拍发票、拍文档）
从复杂背景中提取规则区域（如车牌识别）
图像投影变换、鸟瞰视图生成

本文将通过一个完整案例，带你从图像读取、轮廓检测，到透视变换、结果后处理，逐步完成发票/文件的自动矫正。

图片准备：

1. 导入依赖与辅助函数

import numpy as np
import cv2

我们首先导入 NumPy 和 OpenCV，这是图像处理的核心库。

图像显示函数

def cv_show(name, img):"""显示图像"""cv2.imshow(name, img)cv2.waitKey(0)

cv2.imshow 用于显示图像窗口
cv2.waitKey(0) 表示等待键盘输入后才关闭窗口，便于逐步观察每一步结果

2. 坐标排序函数：`order_points`

def order_points(pts):rect = np.zeros((4, 2), dtype="float32")s = pts.sum(axis=1)rect[0] = pts[np.argmin(s)]  # 左上角点 (x+y最小)rect[2] = pts[np.argmax(s)]  # 右下角点 (x+y最大)diff = np.diff(pts, axis=1)rect[1] = pts[np.argmin(diff)]  # 右上角 (y-x最小)rect[3] = pts[np.argmax(diff)]  # 左下角 (y-x最大)return rect

在透视变换中，必须保证四个点的顺序一致（左上、右上、右下、左下），否则会导致映射结果错误。
该函数通过：

x+y 最小 → 左上角
x+y 最大 → 右下角
y-x 最小 → 右上角
y-x 最大 → 左下角
来对四点坐标排序，保证后续计算稳定。

3. 核心函数：透视变换

def four_point_transform(image,pts):rect = order_points(pts)(tl, tr, br, bl) = rectwidthA = np.sqrt(((br[0] - bl[0]) ** 2) + ((br[1] - bl[1]) ** 2))widthB = np.sqrt(((tr[0] - tl[0]) ** 2) + ((tr[1] - tl[1]) ** 2))maxWidth = max(int(widthA), int(widthB))heightA = np.sqrt(((tr[0] - br[0]) ** 2) + ((tr[1] - br[1]) ** 2))heightB = np.sqrt(((tl[0] - bl[0]) ** 2) + ((tl[1] - bl[1]) ** 2))maxHeight = max(int(heightA), int(heightB))dst = np.array([[0, 0], [maxWidth - 1, 0],[maxWidth - 1, maxHeight - 1],[0, maxHeight - 1]], dtype="float32")M = cv2.getPerspectiveTransform(rect, dst)warped = cv2.warpPerspective(image, M, (maxWidth, maxHeight))return warped

步骤解析：

排序四点：保证输入顺序一致
计算目标矩形宽高：根据两条对边长度取最大值，保证变换后不丢失内容
构建目标点集：从 (0,0) 开始构建新的矩形顶点
求取透视变换矩阵：cv2.getPerspectiveTransform()
执行透视变换：cv2.warpPerspective() 输出矫正后的图像

resize 函数等比例缩放（保持宽高比）

只需指定 width 或 height 中的一个，自动计算另一个

def resize(image, width=None, height=None, inter=cv2.INTER_AREA):dim = None(h, w) = image.shape[:2]if width is None and height is None:return imageif width is None:r = height / float(h)dim = (int(w * r), height)else:r = width / float(w)dim = (width, int(h * r))resized = cv2.resize(image, dim, interpolation=inter)return resized

4. 图像预处理与轮廓检测

image = cv2.imread('order.jpg')
cv_show('image',image)# 缩小图像，便于处理
ratio = image.shape[0]/500
orig = image.copy()
image = resize(orig,height=500)
cv_show('1',image)print('STEP 1: 轮廓检测')
gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)  # 转灰度
# 自动阈值二值化，突出目标区域
edged = cv2.threshold(gray, 0, 255, cv2.THRESH_BINARY | cv2.THRESH_OTSU)[1]
# 寻找所有轮廓
cnts = cv2.findContours(edged.copy(), cv2.RETR_LIST, cv2.CHAIN_APPROX_SIMPLE)[-2]
# 绘制所有轮廓，便于可视化
image_contours = cv2.drawContours(image.copy(), cnts, -1, (0, 0, 255), 1)
cv_show('image_contours', image_contours)

这里先将图像灰度化、二值化，再使用 cv2.findContours 获取所有轮廓。然后在图像上绘制轮廓，便于确认检测效果。

5. 找到最大轮廓（目标区域）

# 找到最大轮廓并逼近为四边形
print("STEP 2: 获取最大轮廓")
screenCnt = sorted(cnts, key=cv2.contourArea, reverse=True)[0]  # 按面积排序，取最大
peri = cv2.arcLength(screenCnt, True)  # 计算周长
screenCnt = cv2.approxPolyDP(screenCnt, 0.05 * peri, True)  # 多边形逼近，保留主要拐点
image_contour = cv2.drawContours(image.copy(), [screenCnt], -1, (0, 0, 255), 2)
cv_show('image_contour', image_contour)

sorted(..., key=cv2.contourArea, reverse=True) 按面积排序，取最大轮廓（一般就是发票/文件区域）
cv2.approxPolyDP 多边形逼近，得到近似四边形轮廓

6. 透视矫正 & 后处理

warped = four_point_transform(orig,screenCnt.reshape(4,2)*ratio)
cv2.imwrite('invoice_new.jpg',warped)
cv_show('xx',warped)# 转灰度并二值化
gray1=cv2.cvtColor(warped,cv2.COLOR_BGR2GRAY)
warped1=cv2.threshold(gray1,120,255,cv2.THRESH_BINARY)[1]
warped1=resize(warped1,height=500)
cv_show('warped1',warped1)# 腐蚀操作，去除噪点
kernel=np.ones((2,2),np.uint8)
erosion = cv2.erode(warped1,kernel,iterations=1)
cv_show('erosion',erosion)# 旋转图像
rotated_image1 =np.rot90(erosion, k=1)
cv_show('rotated_image1',rotated_image1)

后处理步骤：