如何用Python的OpenCV绘制均匀分布的虚线/点线图？

要解决OpenCV中虚线间隔不均的问题，核心是基于整条折线的总长度来均匀分配虚实段，而非逐段处理。以下是具体实现方案：

核心思路 #

1. 先计算整条折线的总长度，同时记录每个顶点在总长度中的累计位置
2. 根据指定的间隔（虚实段的周期），在总长度上均匀生成实线段的起始/结束位置
3. 通过线性插值，将这些长度位置映射回图像坐标，最后逐段绘制实线

实现代码 #

import cv2
import numpy as np

def draw_uniform_dashed_line(img, points, color, dash_period=10, dash_on=5, thickness=1):
    # 转换点为numpy数组，确保格式正确
    points = np.array(points, dtype=np.float32)
    if len(points) < 2:
        return img  # 点数量不足，直接返回原图
    
    # 计算每段线段的长度和累计长度
    segment_lengths = np.sqrt(np.sum(np.diff(points, axis=0)**2, axis=1))
    cumulative_lengths = np.concatenate([[0], np.cumsum(segment_lengths)])
    total_length = cumulative_lengths[-1]
    
    # 如果总长度小于最小显示长度，直接画实线
    if total_length < dash_on:
        cv2.polylines(img, [points.astype(np.int32)], isClosed=False, color=color, thickness=thickness)
        return img
    
    # 生成所有实线段的起始和结束长度位置
    current_pos = 0.0
    segments = []
    while current_pos < total_length:
        seg_start = current_pos
        seg_end = min(current_pos + dash_on, total_length)
        segments.append((seg_start, seg_end))
        current_pos += dash_period
    
    # 将长度位置映射为图像坐标
    def get_point_at_length(length):
        # 找到对应的线段
        idx = np.searchsorted(cumulative_lengths, length) - 1
        if idx >= len(segment_lengths):
            return points[-1]
        # 计算在线段内的相对位置
        seg_start_len = cumulative_lengths[idx]
        seg_rel_pos = (length - seg_start_len) / segment_lengths[idx]
        # 线性插值得到坐标
        return points[idx] + seg_rel_pos * (points[idx+1] - points[idx])
    
    # 绘制每一段实线
    for start_len, end_len in segments:
        start_pt = tuple(get_point_at_length(start_len).astype(np.int32))
        end_pt = tuple(get_point_at_length(end_len).astype(np.int32))
        cv2.line(img, start_pt, end_pt, color, thickness)
    
    return img

# 测试示例
if __name__ == "__main__":
    # 创建空白图像
    img = np.zeros((500, 500, 3), dtype=np.uint8)
    # 定义折线点
    line_points = [(50, 50), (200, 300), (400, 100), (450, 450)]
    # 绘制均匀虚线
    draw_uniform_dashed_line(img, line_points, (0, 255, 0), dash_period=20, dash_on=10, thickness=2)
    # 显示结果
    cv2.imshow("Uniform Dashed Line", img)
    cv2.waitKey(0)
    cv2.destroyAllWindows()

关键细节说明 #

• 累计长度计算：通过np.diff和np.cumsum快速计算每段线段长度和总长度，确保能追踪任意位置对应的线段
• 线性插值：当目标位置不在顶点上时，通过相对比例计算坐标，保证虚线过渡平滑
• 边缘处理：当总长度小于实线段长度时，直接绘制实线；当剩余长度不足一个周期时，自动截断到终点，避免出现过短的无效线段

内容的提问来源于stack exchange，提问作者Erik Hulmák