如何从cv2.findContours提取数组元素，获取线段端点计算斜率以纠偏文档？

先明确cv2.findContours的返回值差异：

• 若使用OpenCV 4.x及以上，返回contours, hierarchy两个值
• 若使用OpenCV 3.x，返回image, contours, hierarchy三个值

其中contours是一个列表，每个元素对应一个轮廓。每个轮廓是形状为(N, 1, 2)的numpy.ndarray：

• N是该轮廓包含的像素点数量
• 1是OpenCV格式要求的冗余维度
• 2代表每个点的(x, y)坐标

你可以把单个轮廓转换成更直观的(N, 2)格式：

contour = contours[0]  # 你的图像只有一条线段，取第一个轮廓即可
points = contour.reshape(-1, 2)  # 去掉冗余维度，得到(N,2)的坐标数组

提取线段端点的两种实用方法 #

方法1：多边形近似（推荐）

利用cv2.approxPolyDP把线段轮廓近似为两个端点，适合大多数情况：

import cv2
import numpy as np

# 读取并二值化图像（确保线段是前景）
img = cv2.imread('your_image.png', 0)
_, binary_img = cv2.threshold(img, 127, 255, cv2.THRESH_BINARY_INV)

# 查找轮廓（用RETR_EXTERNAL只找最外层轮廓）
contours, hierarchy = cv2.findContours(binary_img, cv2.RETR_EXTERNAL, cv2.CHAIN_APPROX_SIMPLE)

# 取目标轮廓
contour = contours[0]
# 计算近似多边形：epsilon设为轮廓周长的1%，closed=False因为线段不是闭合图形
epsilon = 0.01 * cv2.arcLength(contour, closed=False)
approx = cv2.approxPolyDP(contour, epsilon, closed=False)
# 转换为端点数组
end_points = approx.reshape(-1, 2)

print("线段端点：", end_points)

方法2：极值点提取

如果线段足够笔直，直接找x坐标的极值点即可：

points = contour.reshape(-1, 2)
left_point = points[np.argmin(points[:, 0])]  # x最小的左端点
right_point = points[np.argmax(points[:, 0])]  # x最大的右端点
end_points = np.array([left_point, right_point])

计算斜率与图像纠偏 #

拿到端点后直接计算斜率和角度，完成纠偏：

(x1, y1), (x2, y2) = end_points
slope = (y2 - y1) / (x2 - x1)
# 转换为角度（弧度→角度）
angle_deg = np.degrees(np.arctan(slope))

# 执行图像纠偏
h, w = img.shape[:2]
center = (w // 2, h // 2)
# 生成旋转矩阵，负角度实现顺时针纠偏
M = cv2.getRotationMatrix2D(center, -angle_deg, 1.0)
corrected_img = cv2.warpAffine(img, M, (w, h), flags=cv2.INTER_CUBIC, borderMode=cv2.BORDER_REPLICATE)

内容的提问来源于stack exchange，提问作者Chris