C 语言

关注公众号 jb51net

关闭
首页 > 软件编程 > C 语言 > OpenCV模板匹配

C/C++的OpenCV实现模板匹配的基础实现与优化

作者：whoarethenext

模板匹配（Template Matching）是计算机视觉中的一个基础且广泛应用的技术,本文将为大家详细介绍一下基于OpenCV的模板匹配基础实现与优化,有需要的可以参考下

模板匹配（Template Matching）是计算机视觉中的一个基础且广泛应用的技术。它的主要任务是在一张大图中（源图像）寻找并定位一个特定的小图（模板图像）出现的位置。

本篇文章将引导你完成两部分内容：

环境准备

在开始之前，请确保你已经准备好以下环境：

一个 C++ 编译器 (例如 G++, Clang, MSVC)。

OpenCV 库已经安装并配置好。

准备两张图片：

Part 1: 基础模板匹配

这是最直接的实现方式，通过在源图像上滑动模板，逐一比较像素值来计算相似度。

核心逻辑

加载源图像和模板图像。

使用 cv::matchTemplate() 函数计算模板在源图像上每个可能位置的相似度得分，并将结果存入一个结果矩阵。

使用 cv::minMaxLoc() 函数在结果矩阵中找到最优点（最大值或最小值的位置，取决于所用算法）。

在源图像上用矩形框标出这个最优点。

C++ 代码示例

#include <iostream>
#include <opencv2/opencv.hpp>

int main() {
    // 1. 加载源图像和模板图像
    cv::Mat srcImage = cv::imread("source_image.jpg", cv::IMREAD_COLOR);
    cv::Mat templateImage = cv::imread("template_image.jpg", cv::IMREAD_COLOR);

    if (srcImage.empty() || templateImage.empty()) {
        std::cerr << "错误: 无法加载图像!" << std::endl;
        return -1;
    }

    // 2. 创建用于存放结果的矩阵
    int result_cols = srcImage.cols - templateImage.cols + 1;
    int result_rows = srcImage.rows - templateImage.rows + 1;
    cv::Mat resultImage;
    resultImage.create(result_rows, result_cols, CV_32FC1);

    // 3. 执行模板匹配 (使用归一化相关系数法)
    cv::matchTemplate(srcImage, templateImage, resultImage, cv::TM_CCOEFF_NORMED);

    // 4. 找到最佳匹配位置
    double minVal, maxVal;
    cv::Point minLoc, maxLoc;
    cv::minMaxLoc(resultImage, &minVal, &maxVal, &minLoc, &maxLoc);

    // 对于 TM_CCOEFF_NORMED 方法，最佳匹配点是最大值所在的位置
    cv::Point matchLoc = maxLoc;

    // 5. 在源图像上绘制矩形框来标记匹配区域
    cv::rectangle(srcImage, matchLoc,
                  cv::Point(matchLoc.x + templateImage.cols, matchLoc.y + templateImage.rows),
                  cv::Scalar(0, 255, 0), 2); // 绿色矩形

    // 6. 显示结果
    cv::imshow("基础匹配结果", srcImage);
    cv::imshow("模板图像", templateImage);

    cv::waitKey(0);
    return 0;
}

Part 2: 使用直方图均衡化提升识别率

基础匹配在光照变化、对比度低等复杂场景下，识别率可能会下降。为了解决这个问题，我们可以在匹配前对图像进行预处理，直方图均衡化就是一种简单而高效的增强对比度的方法。

优化思路

通过 cv::equalizeHist 函数增强源图像和模板图像的对比度，让图像中的细节和轮廓更加突出，从而让模板匹配算法能更稳定地工作。

优化后的 C++ 代码

#include <iostream>
#include <opencv2/opencv.hpp>
#include <opencv2/imgproc.hpp> // 包含图像处理模块头文件

int main() {
    // 1. 加载原始的彩色图像
    cv::Mat srcColorImage = cv::imread("source_image.jpg", cv::IMREAD_COLOR);
    cv::Mat templateColorImage = cv::imread("template_image.jpg", cv::IMREAD_COLOR);

    if (srcColorImage.empty() || templateColorImage.empty()) {
        std::cerr << "错误: 无法加载图像!" << std::endl;
        return -1;
    }

    // --- 图像预处理步骤 ---

    // 2. 将图像转换为灰度图
    cv::Mat srcGray, templateGray;
    cv::cvtColor(srcColorImage, srcGray, cv::COLOR_BGR2GRAY);
    cv::cvtColor(templateColorImage, templateGray, cv::COLOR_BGR2GRAY);

    // 3. 对灰度图像进行直方图均衡化
    cv::Mat srcEqualized, templateEqualized;
    cv::equalizeHist(srcGray, srcEqualized);
    cv::equalizeHist(templateGray, templateEqualized);

    // --- 预处理结束 ---


    // 4. 创建结果矩阵
    int result_cols = srcEqualized.cols - templateEqualized.cols + 1;
    int result_rows = srcEqualized.rows - templateEqualized.rows + 1;
    cv::Mat resultImage;
    resultImage.create(result_rows, result_cols, CV_32FC1);

    // 5. 在【均衡化后】的图像上执行模板匹配
    cv::matchTemplate(srcEqualized, templateEqualized, resultImage, cv::TM_CCOEFF_NORMED);

    // 6. 找到最佳匹配位置
    double minVal, maxVal;
    cv::Point minLoc, maxLoc;
    cv::minMaxLoc(resultImage, &minVal, &maxVal, &minLoc, &maxLoc);
    cv::Point matchLoc = maxLoc;

    // 7. 在【原始彩色 图像】上绘制矩形框
    cv::rectangle(srcColorImage, matchLoc,
                  cv::Point(matchLoc.x + templateColorImage.cols, matchLoc.y + templateColorImage.rows),
                  cv::Scalar(0, 0, 255), 2); // 红色矩形以示区别

    // 8. 显示所有相关图像
    cv::imshow("最终匹配结果", srcColorImage);
    cv::imshow("原始模板", templateColorImage);
    cv::imshow("均衡化后的模板", templateEqualized); // 显示处理效果

    cv::waitKey(0);
    return 0;
}

如何编译和运行

无论你选择哪一个版本的代码，编译和运行的步骤都是相同的。将代码保存为 C++ 文件（例如 matcher.cpp），并和你的图片放在同一目录下。

编译 (以 Linux/macOS 的 g++ 为例)

推荐使用 pkg-config 工具自动链接 OpenCV 库，这能省去手动指定路径的麻烦。

g++ matcher.cpp -o matcher `pkg-config --cflags --libs opencv4`

如果你的系统没有 pkg-config 或者 OpenCV 版本不同，你可能需要手动指定头文件和库的路径：

# 注意: /path/to/opencv/ 需要替换为你的实际安装路径
g++ matcher.cpp -o matcher -I/path/to/opencv/include -L/path/to/opencv/lib -lopencv_core -lopencv_highgui -lopencv_imgproc -lopencv_imgcodecs

运行

编译成功后，会生成一个名为 matcher 的可执行文件。

./matcher

程序将自动运行，并弹出窗口显示匹配结果。

总结

本文展示了从一个基础的模板匹配实现到一个经过优化的稳健方案。

到此这篇关于C/C++的OpenCV实现模板匹配的基础实现与优化的文章就介绍到这了,更多相关OpenCV模板匹配内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！

您可能感兴趣的文章:
阅读全文