基于opencv模板匹配的目标识别方法

简介: 因为pcl的点云模板匹配遇到了各种困难,暂时先用opencv的模板匹配函数做一个简单的焊缝识别,看看效果。此方法的缺陷就在于物体和相机位置必须固定,只允许微小位移,否则数据将失效。

因为pcl的点云模板匹配遇到了各种困难,暂时先用opencv的模板匹配函数做一个简单的焊缝识别,看看效果。此方法的缺陷就在于物体和相机位置必须固定,只允许微小位移,否则数据将失效。

1

什么是模板匹配?
模板匹配是一种用于查找与模板图像(补丁)匹配(类似)的图像区域的技术。
虽然补丁必须是一个矩形,可能并不是所有的矩形都是相关的。在这种情况下,可以使用掩模来隔离应该用于找到匹配的补丁部分。

它是如何工作的?

  • 我们需要两个主要组件:
  1. 源图像(I):我们期望找到与模板图像匹配的图像
  2. 模板图像(T):将与模板图像进行比较的补丁图像


    img_5e308d8706d214f68bf9735be72a5e48.png
    源图像--->模板图像--->匹配结果
  • 要识别匹配区域,我们必须通过滑动来比较模板图像与源图像
  • 通过滑动,我们的意思是一次移动补丁一个像素(从左到右,从上到下)。在每个位置,计算度量,以便它表示在该位置处的匹配的“好”还是“坏”(或者与图像的特定区域相似)。

2

OpenCV在函数matchTemplate()中实现模板匹配。可用的方法有以下6种(我采取的上第二种):

img_c52cd15066f0bcf88be0c7847b758643.png

img_3770cde6726a304bab8b823daeb6ca40.png


3

img_d0b4d08d231e3d409d7a7ed3acde6f7c.png
源图像(工件)

img_391cbd509e4e78442fa8c48653b72c9d.png
模板图像(焊缝)
img_677af3ec660ceef3f03bbc67dc7efe49.png
匹配结果
img_05565574a8dd71ac77d078d6bce9b570.png
焊缝锁定

img_8ea375bfbd79de10da0a8b0488af4e27.png
端点空间坐标输出

然后通过匹配得到焊缝位置,使用getPointXYZ将像素坐标(u,v)转化为相机坐标(x,y,z)。从而得到焊缝的空间坐标,后续将xyz传给机械臂,为手眼标定做准备。


4

关键代码

其中因为matchTemplate的输入图像要求是三通道,而我得到的rgbd为四通道,需要使用cvtColor将四通道改为三通道(cv::cvtColor(rgbd, rgbd, COLOR_BGRA2BGR);)很关键,当时纠结了好久。

    Mat rgbmat, depthmat, irmat, rgbd, model, result;
    cv::namedWindow("registered", WND_PROP_ASPECT_RATIO);
    //cv::namedWindow("result", WND_PROP_ASPECT_RATIO);
    model = imread("model.png");
    float x1, y1, z1, x2, y2, z2;
    Point p1, p2;

    while(!protonect_shutdown)
    {
        listener.waitForNewFrame(frames);
        libfreenect2::Frame *rgb = frames[libfreenect2::Frame::Color];
        libfreenect2::Frame *ir = frames[libfreenect2::Frame::Ir];
        libfreenect2::Frame *depth = frames[libfreenect2::Frame::Depth];

        cv::Mat(rgb->height, rgb->width, CV_8UC4, rgb->data).copyTo(rgbmat);
        cv::Mat(ir->height, ir->width, CV_32FC1, ir->data).copyTo(irmat);
        cv::Mat(depth->height, depth->width, CV_32FC1, depth->data).copyTo(depthmat);
        registration->apply(rgb, depth, &undistorted, &registered, true, &depth2rgb);
        cv::Mat(registered.height, registered.width, CV_8UC4, registered.data).copyTo(rgbd);


        cv::cvtColor(rgbd, rgbd, COLOR_BGRA2BGR);

        int result_cols = rgbd.cols - model.cols + 1;
        int result_rows = rgbd.rows - model.rows + 1;
        result.create(result_cols, result_rows, CV_32FC1);

        matchTemplate(rgbd, model, result, CV_TM_SQDIFF_NORMED);//这里我们使用的匹配算法是标准平方差匹配 method=CV_TM_SQDIFF_NORMED,数值越小匹配度越好
        normalize(result, result, 0, 1, NORM_MINMAX, -1, Mat());

        double minVal = -1;
        double maxVal;
        Point minLoc;
        Point maxLoc;
        Point matchLoc;
        //cout << "匹配度:" << minVal << endl;
        minMaxLoc(result, &minVal, &maxVal, &minLoc, &maxLoc, Mat());
        //cout << "匹配度:" << minVal << endl;

        matchLoc = minLoc;
        p1 = Point(matchLoc.x+5, matchLoc.y+8);
        p2 = Point(matchLoc.x-4 + model.cols, matchLoc.y+13);

        //rectangle(rgbd, matchLoc, Point(matchLoc.x + model.cols, matchLoc.y + model.rows), Scalar(0, 255, 0), 2, 8, 0);
        registration->getPointXYZ(&undistorted, p1.y, p1.x, x1, y1, z1 );
        registration->getPointXYZ(&undistorted, p2.y, p2.x, x2, y2, z2 );
        circle(rgbd, p1, 1, Scalar(255,255,0), 2);
        circle(rgbd, p2, 1, Scalar(255,255,0), 2);
        line(rgbd, p1, p2, Scalar(0,255,0));
        cv::imshow("registered", rgbd);
        //cv::imshow("result", result);
        //cout << matchLoc.x <<"  "<<  matchLoc.y<<endl;
        cout <<"p1:"<< x1 <<"  "<< y1 <<"  "<< z1 << endl;
        cout <<"p2:"<< x2 <<"  "<< y2 <<"  "<< z2 << endl;
        
        int key = cv::waitKey(1);
        protonect_shutdown = protonect_shutdown || (key > 0 && ((key & 0xFF) == 27)); // shutdown on escape
 
        listener.release(frames);
    }
目录
相关文章
|
16天前
|
JSON 数据格式 计算机视觉
Opencv实用笔记(一): 获取并绘制JSON标注文件目标区域(可单独保存目标小图)
本文介绍了如何使用OpenCV和Python根据JSON标注文件获取并绘制目标区域,同时可将裁剪的图像单独保存。通过示例代码,展示了如何读取图片路径、解析JSON标注、绘制标注框并保存裁剪图像的过程。此外,还提供了相关的博客链接,供读者进一步学习。
25 0
|
3月前
|
存储 编解码 API
python多种方法压缩图片,opencv、PIL、tinypng、pngquant压缩图片
python多种方法压缩图片,opencv、PIL、tinypng、pngquant压缩图片
188 1
|
3月前
|
算法 定位技术 vr&ar
一文了解PnP算法,python opencv中的cv2.solvePnP()的使用,以及使用cv2.sovlePnP()方法标定相机和2D激光雷达
一文了解PnP算法,python opencv中的cv2.solvePnP()的使用,以及使用cv2.sovlePnP()方法标定相机和2D激光雷达
372 0
一文了解PnP算法,python opencv中的cv2.solvePnP()的使用,以及使用cv2.sovlePnP()方法标定相机和2D激光雷达
|
4月前
|
机器学习/深度学习 传感器 算法
OpenCV4工业缺陷检测的六种方法
OpenCV4工业缺陷检测的六种方法
|
4月前
|
算法 计算机视觉 索引
python---OpenCv(二),背景分离方法较有意思
python---OpenCv(二),背景分离方法较有意思
|
5月前
|
运维 算法 计算机视觉
【Qt&OpenCV 图像的模板匹配 matchTemplate/minMaxLoc】
【Qt&OpenCV 图像的模板匹配 matchTemplate/minMaxLoc】
72 1
|
4月前
|
计算机视觉 Python
opencv 处理图像去噪的几种方法学习
OpenCV 提供了多种图像去噪的方法,以下是一些常见的去噪技术以及相应的 Python 代码示例: 均值滤波:使用像素邻域的灰度均值代替该像素的值。
58 0
|
5月前
|
算法 计算机视觉
如何判断点在多边形内部:OpenCV--cv2.pointPolygonTest()方法详解
如何判断点在多边形内部:OpenCV--cv2.pointPolygonTest()方法详解
|
6月前
|
计算机视觉
OpenCV 模板匹配
OpenCV 模板匹配
59 1
|
6月前
|
算法 API 计算机视觉
【OpenCV】- 模板匹配(浩瀚星空只为寻找那一抹明月)
【OpenCV】- 模板匹配(浩瀚星空只为寻找那一抹明月)