图像轮廓匹配
图像的几何矩和Hu矩
- 几何矩
几何矩是由Hu(Visual pattern recognition by moment invariants)在1962年提出的,具有平移、旋转和尺度不变性。 定义如下:
- Hu矩
图像轮廓匹配计算
- 基于Hu矩轮廓匹配
- 相关API:
moments
和HuMoments
和matchShapes
//计算几何矩
CV_EXPORTS_W Moments moments( InputArray array, bool binaryImage = false );
//计算Hu矩
CV_EXPORTS_W void HuMoments( const Moments& m, OutputArray hu );
//轮廓匹配
CV_EXPORTS_W double matchShapes( InputArray contour1, InputArray contour2,
int method, double parameter );
其中 matchShapes
的method有三种:CONTOURS_MATCH_I1 、CONTOURS_MATCH_I2 、CONTOURS_MATCH_I3
,返回值越小说明轮廓相似程度越高!
//获取图像轮廓方法
void contour_info(Mat &image, vector<vector<Point>>&contours);
//图像轮廓匹配
int main(int argc, char** argv) {
Mat src1 = imread("F:/code/images/abc.png");
Mat src2 = imread("F:/code/images/a5.png");
CV_Assert(!src1.empty());
CV_Assert(!src2.empty());
imshow("input1", src1);
imshow("input2", src2);
vector<vector<Point>>contours1;
vector<vector<Point>>contours2;
//获取轮廓
contour_info(src1, contours1);
contour_info(src2, contours2);
//轮廓Hu计算
//将src2中的一个轮廓作为模板,在src1中的很多轮廓中查找匹配
//src2
Moments mm2 = moments(contours2[0]);
Mat Hu2;
HuMoments(mm2, Hu2);
//src1
for (size_t t = 0; t < contours1.size(); t++) {
Moments mm1 = moments(contours1[t]);
//可以获取轮廓的中心位置
double cx = mm1.m10 / mm1.m00;
double cy = mm1.m01 / mm1.m00;
circle(src1, Point(cx, cy), 3, Scalar(255, 0, 0), 2, 8, 0);
Mat Hu1;
HuMoments(mm1, Hu1);
double dist = matchShapes(Hu1, Hu2, CONTOURS_MATCH_I1, 0);
printf("matched distance value: %.2f\n", dist);
if (dist < 1) {
drawContours(src1, contours1, t, Scalar(0, 0, 255), 2, 8);
}
}
imshow("match contours demo", src1);
waitKey(0);
destroyAllWindows();
return 0;
}
void contour_info(Mat& image, vector<vector<Point>>& contours) {
Mat dst;
//高斯滤波
GaussianBlur(image, dst, Size(3, 3), 0);
//得到灰度图
Mat gray, binary;
cvtColor(dst, gray, COLOR_RGB2GRAY);
//得到二值图
threshold(gray, binary, 0, 255, THRESH_BINARY | THRESH_OTSU);
//发现轮廓
vector<Vec4i> hierarchy;//层次信息
findContours(binary, contours, hierarchy, RETR_EXTERNAL, CHAIN_APPROX_SIMPLE, Point());//RETR_EXTERNAL 获取最大的轮廓
printf("%llu\n", contours.size());
}