基于Python的施工图与竣工图对比小工具开发方案

基于Python的施工图与竣工图对比小工具开发方案

---

一、引言

在工程建设领域，施工图与竣工图的对比是项目验收的关键环节。传统人工对比方式效率低、易出错，本文基于Python开发一款自动化对比工具，支持快速检测图纸差异并生成可视化报告，大幅提升工程审核效率。

二、技术方案概述

核心功能

1. 图像预处理：灰度化、降噪，统一图纸格式
2. 特征匹配：基于ORB算法提取图纸特征点并匹配
3. 差异检测：通过单应性变换对齐图纸，计算像素级差异
4. 报告生成：生成含差异标记的PDF报告，可视化展示修改区域

三、环境准备与依赖安装

1. 系统要求

• Python 3.9+（推荐3.10）
• Windows/macOS/Linux（本文以Windows为例）

2. 安装依赖库

pip install opencv-python matplotlib reportlab pyinstaller

• opencv-python：图像处理与特征检测
• matplotlib：临时图像保存（用于报告生成）
• reportlab：生成专业PDF报告
• pyinstaller：后期打包EXE文件

四、核心模块架构设计

四层架构设计

关键技术点

1. ORB特征检测：结合FAST关键点检测和BRIEF描述子，兼具速度与精度
2. 单应性变换（Homography）：通过findHomography计算图纸变换矩阵，解决视角/缩放差异
3. 像素级差异计算：使用absdiff计算灰度差异，二值化后高亮显示修改区域

五、详细开发步骤

1. 图像预处理模块

import cv2

def preprocess_image(image_path):
    """读取图像并转换为灰度图"""
    image = cv2.imread(image_path)
    # BGR转灰度（OpenCV默认BGR通道）
    gray = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
    # 可选：高斯模糊降噪
    # gray = cv2.GaussianBlur(gray, (5, 5), 0)
    return gray

2. 特征提取与匹配模块

def extract_and_match_features(img1, img2):
    """ORB特征提取与BF匹配"""
    orb = cv2.ORB_create()  # 初始化ORB检测器
    # 检测关键点与描述符
    kp1, des1 = orb.detectAndCompute(img1, None)
    kp2, des2 = orb.detectAndCompute(img2, None)
    # 暴力匹配（Hamming距离）
    bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)
    matches = bf.match(des1, des2)
    # 按距离排序（距离越小匹配度越高）
    matches = sorted(matches, key=lambda x: x.distance)
    return kp1, kp2, matches

3. 差异检测模块

def detect_differences(img1, img2, kp1, kp2, matches):
    """基于单应性变换的差异检测"""
    # 取前10个最佳匹配点（实际项目建议动态调整）
    good_matches = matches[:10]
    src_pts = np.float32([kp1[m.queryIdx].pt for m in good_matches]).reshape(-1,1,2)
    dst_pts = np.float32([kp2[m.trainIdx].pt for m in good_matches]).reshape(-1,1,2)
    # 计算单应性矩阵（RANSAC算法抗噪声）
    M, _ = cv2.findHomography(src_pts, dst_pts, cv2.RANSAC, 5.0)
    # 透视变换对齐图纸
    aligned_img = cv2.warpPerspective(img1, M, (img2.shape[1], img2.shape[0]))
    # 计算灰度差异（绝对值）
    diff = cv2.absdiff(aligned_img, img2)
    # 二值化处理（差异阈值30可调）
    _, binary_diff = cv2.threshold(diff, 30, 255, cv2.THRESH_BINARY)
    return binary_diff

4. 报告生成模块

from reportlab.lib.pagesizes import letter
from reportlab.pdfgen import canvas

def generate_report(img1, img2, diff, report_path):
    """生成含对比图的PDF报告"""
    c = canvas.Canvas(report_path, pagesize=letter)
    width, height = letter  # A4纸尺寸（210mm=792pt, 297mm=1098pt）
    
    # 保存临时图像（需转换为RGB格式，OpenCV默认BGR）
    cv2.imwrite('img1.png', cv2.cvtColor(img1, cv2.COLOR_GRAY2BGR))
    cv2.imwrite('img2.png', cv2.cvtColor(img2, cv2.COLOR_GRAY2BGR))
    cv2.imwrite('diff.png', cv2.cvtColor(diff, cv2.COLOR_GRAY2BGR))
    
    # 绘制对比图（左：施工图，中：竣工图，下：差异图）
    c.drawImage('img1.png', 50, 400, width=300, height=400)  # 左上角坐标(50,400)
    c.drawImage('img2.png', 350, 400, width=300, height=400)
    c.drawImage('diff.png', 50, 50, width=600, height=300)
    
    # 添加标题与说明
    c.setFont('Helvetica-Bold', 20)
    c.drawString(200, 1000, '施工图 vs 竣工图对比分析报告')
    c.setFont('Helvetica', 14)
    c.drawString(100, 380, '施工图（原图）')
    c.drawString(400, 380, '竣工图（对比图）')
    c.drawString(150, 30, '差异区域（白色为修改部分）')
    
    c.save()  # 保存PDF

六、完整代码实现

import cv2
from reportlab.lib.pagesizes import letter
from reportlab.pdfgen import canvas

def preprocess_image(image_path):
    image = cv2.imread(image_path)
    return cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)

def extract_and_match_features(img1, img2):
    orb = cv2.ORB_create()
    kp1, des1 = orb.detectAndCompute(img1, None)
    kp2, des2 = orb.detectAndCompute(img2, None)
    bf = cv2.BFMatcher(cv2.NORM_HAMMING, crossCheck=True)
    return bf.match(des1, des2)

def detect_differences(img1, img2, matches):
    good_matches = matches[:10]
    src_pts = np.float32([img1[kp.queryIdx].pt for kp in good_matches]).reshape(-1,1,2)
    dst_pts = np.float32([img2[kp.trainIdx].pt for kp in good_matches]).reshape(-1,1,2)
    M, _ = cv2.findHomography(src_pts, dst_pts, cv2.RANSAC, 5.0)
    aligned = cv2.warpPerspective(img1, M, (img2.shape[1], img2.shape[0]))
    diff = cv2.absdiff(aligned, img2)
    _, binary = cv2.threshold(diff, 30, 255, cv2.THRESH_BINARY)
    return binary

def generate_report(img1, img2, diff, report_path):
    cv2.imwrite('temp1.png', cv2.cvtColor(img1, cv2.COLOR_GRAY2BGR))
    cv2.imwrite('temp2.png', cv2.cvtColor(img2, cv2.COLOR_GRAY2BGR))
    cv2.imwrite('temp_diff.png', cv2.cvtColor(diff, cv2.COLOR_GRAY2BGR))
    
    c = canvas.Canvas(report_path, pagesize=letter)
    c.drawImage('temp1.png', 50, 500, 300, 400)
    c.drawImage('temp2.png', 350, 500, 300, 400)
    c.drawImage('temp_diff.png', 50, 100, 600, 300)
    c.setFont('Helvetica-Bold', 20).drawString(250, 1050, '图纸对比报告')
    c.save()

def main(construction_path, asbuilt_path, report_path):
    img1 = preprocess_image(construction_path)
    img2 = preprocess_image(asbuilt_path)
    matches = extract_and_match_features(img1, img2)
    diff = detect_differences(img1, img2, matches)
    generate_report(img1, img2, diff, report_path)

if __name__ == "__main__":
    main("construction.png", "asbuilt.png", "comparison_report.pdf")

七、本地化打包EXE文件（Windows平台）

1. 安装PyInstaller

pip install pyinstaller

2. 执行打包命令

pyinstaller --onefile --windowed drawing_comparison.py

• --onefile：生成单个EXE文件
• --windowed：隐藏命令行窗口（可选）

3. 运行EXE

打包成功后，在dist目录找到drawing_comparison.exe，双击即可运行（无需安装Python环境）

八、使用说明与优化建议

1. 输入要求

• 支持JPG/PNG格式，建议分辨率≥300dpi
• 图纸背景需干净，避免多余标注干扰检测

2. 进阶优化

1. 多尺度检测：增加图像金字塔处理，适应不同比例图纸
2. 边缘检测增强：结合Canny算子预处理，提升复杂线条对比精度
3. 文本差异识别：集成OCR技术（如Tesseract），检测文字修改区域
4. GUI界面：使用PyQt5/Pygame开发图形化界面，提升交互体验

3. 常见问题

• 匹配失败：增加good_matches数量（如前20个匹配点）
• 差异模糊：调整cv2.threshold的阈值参数（当前30，范围0-255）
• PDF乱码：添加中文字体支持（需修改reportlab配置）

九、总结

本文提供了从算法设计到工程落地的完整解决方案，实现了图纸对比的自动化与可视化。工具在简单场景下已能满足需求，复杂工程场景可通过增加机器学习模型（如YOLO目标检测）进一步提升精度。后续可扩展支持DWG/DXF等CAD原生格式，打造更专业的工程文档对比平台。