#include <iostream>
#include <vector>
//Opencv库
#include <opencv.hpp>
void findcontours(cv::Mat &image,std::vector<std::vector<cv::Point>> &contours)
{
cv::Mat gray,binary;
std::vector<cv::Vec4i> hierarchy;
//图像灰度化
cv::cvtColor(image,gray,cv::COLOR_BGR2GRAY);
//图像二值化
cv::threshold(gray,binary,0,255,cv::THRESH_BINARY|cv::THRESH_OTSU);
//寻找轮廓
cv::findContours(binary,contours,hierarchy,0,2);
}
int main()
{
cv::Mat img_ABC = cv::imread("ABC.png");
cv::Mat img_B = cv::imread("ABC.png");
cv::resize(img_B,img_B,cv::Size(),0.5,0.5);
std::vector<std::vector<cv::Point>> contours_1;
std::vector<std::vector<cv::Point>> contours_2;
//寻找轮廓
findcontours(img_ABC,contours_1);
findcontours(img_B,contours_2);
//计算轮廓对应的矩
cv::Moments mm_B = cv::moments(contours_2[0]);
cv::Mat hu_B;
cv::HuMoments(mm_B,hu_B);
//轮廓匹配
for(int n = 0;n < contours_1.size();n++)
{
cv::Moments mm = cv::moments(contours_1[n]);
cv::Mat hu_m;
cv::HuMoments(mm,hu_m);
//Hu矩匹配
double dist;
dist = cv::matchShapes(hu_m,hu_B,cv::CONTOURS_MATCH_I1,0);
if(dist < 1)
{
cv::drawContours(img_ABC,contours_1,n,cv::Scalar(0,0,255),3,0);
}
}
cv::imshow("Test",img_ABC);
cv::waitKey(0);
return 0;
}
Opencv做形状匹配
最新推荐文章于 2024-05-18 20:41:11 发布