图像匹配算法比较与分析

来源:互联网 发布:手机wifi挂机赚钱软件 编辑:程序博客网 时间:2024/05/22 05:32

1.感知哈希(pHash)算法的Opencv实现

均值Hash算法

  1. //均值Hash算法  
  2.  string HashValue(Mat &src)  
  3.  {  
  4.      string rst(64,'\0');  
  5.      Mat img;  
  6.      if(src.channels()==3)  
  7.          cvtColor(src,img,CV_BGR2GRAY);  
  8.     else  
  9.         img=src.clone();  
  10.        /*第一步,缩小尺寸。 
  11.          将图片缩小到8x8的尺寸,总共64个像素,去除图片的细节*/  
  12.   
  13.         resize(img,img,Size(8,8));  
  14.     /* 第二步,简化色彩(Color Reduce)。 
  15.        将缩小后的图片,转为64级灰度。*/  
  16.   
  17.     uchar *pData;  
  18.     for(int i=0;i<img.rows;i++)  
  19.     {  
  20.         pData = img.ptr<uchar>(i);  
  21.         for(int j=0;j<img.cols;j++)  
  22.         {  
  23.             pData[j]=pData[j]/4;            }  
  24.     }  
  25.       
  26.         /* 第三步,计算平均值。 
  27.        计算所有64个像素的灰度平均值。*/  
  28.     int average = mean(img).val[0];  
  29.       
  30.         /* 第四步,比较像素的灰度。 
  31.      将每个像素的灰度,与平均值进行比较。大于或等于平均值记为1,小于平均值记为0*/  
  32.     Mat mask= (img>=(uchar)average);  
  33.       
  34.         /* 第五步,计算哈希值。*/  
  35.     int index = 0;  
  36.     for(int i=0;i<mask.rows;i++)  
  37.     {  
  38.         pData = mask.ptr<uchar>(i);  
  39.         for(int j=0;j<mask.cols;j++)  
  40.         {  
  41.             if(pData[j]==0)  
  42.                 rst[index++]='0';  
  43.             else  
  44.                 rst[index++]='1';  
  45.         }  
  46.     }  
  47.     return rst;  
  48.  }  

pHash算法

  1. //pHash算法  
  2.  string pHashValue(Mat &src)  
  3.  {  
  4.      Mat img ,dst;  
  5.      string rst(64,'\0');  
  6.      double dIdex[64];  
  7.      double mean = 0.0;  
  8.      int k = 0;  
  9.      if(src.channels()==3)  
  10.      {  
  11.          cvtColor(src,src,CV_BGR2GRAY);  
  12.          img = Mat_<double>(src);  
  13.      }     
  14.      else  
  15.      {  
  16.          img = Mat_<double>(src);  
  17.      }       
  18.        
  19.         /* 第一步,缩放尺寸*/  
  20.      resize(img, img, Size(8,8));  
  21.           
  22.         /* 第二步,离散余弦变换,DCT系数求取*/  
  23.      dct(img, dst);   
  24.   
  25.         /* 第三步,求取DCT系数均值(左上角8*8区块的DCT系数)*/  
  26.      for (int i = 0; i < 8; ++i) {  
  27.          for (int j = 0; j < 8; ++j)   
  28.          {  
  29.              dIdex[k] = dst.at<double>(i, j);  
  30.              mean += dst.at<double>(i, j)/64;  
  31.              ++k;  
  32.          }  
  33.      }  
  34.           
  35.         /* 第四步,计算哈希值。*/  
  36.      for (int i =0;i<64;++i)  
  37.      {  
  38.          if (dIdex[i]>=mean)  
  39.          {  
  40.              rst[i]='1';  
  41.          }  
  42.          else  
  43.          {  
  44.              rst[i]='0';  
  45.          }  
  46.      }  
  47.      return rst;  
  48.  }  

汉明距离计算     


 

         通过上面两段代码就可以计算出图像的Hash值,检索的时候一般采用汉明距离来进行判断两幅图像的相似性,一般情况下认为汉明距离小于5,就可以认为两幅图像时相似的。汉明具体计算实现:

[cpp] view plain copy
  1. //汉明距离计算  
  2.  int HanmingDistance(string &str1,string &str2)  
  3.  {  
  4.     if((str1.size()!=64)||(str2.size()!=64))  
  5.         return -1;  
  6.     int difference = 0;  
  7.     for(int i=0;i<64;i++)  
  8.     {  
  9.         if(str1[i]!=str2[i])  
  10.             difference++;  
  11.     }  
  12.     return difference;  
  13.  }  

哈希值海明距离图像匹配应用

#include<iostream>
#include<opencv2/opencv.hpp>
using namespace std;
using namespace cv;
string pHashValue(Mat &src);
int HanmingDistance(string &str1, string &str2);
void main()
{
string pHashValue(Mat &src);
int HanmingDistance(string &str1, string &str2);
Mat srcImage = imread("1-300(1).bmp", 1);
Mat srcImage1 = imread("1-130(1).bmp", 1);
string rst1(64, '\0');
string rst2(64, '\0');
rst1=pHashValue(srcImage);
rst2=pHashValue(srcImage1);
int t;
t=HanmingDistance(rst1, rst2);
cout << "图一与图二的海明距离为=" << t << endl;
}
//pHash算法  
string pHashValue(Mat &src)
{
Mat img, dst;
string rst(64, '\0');
double dIdex[64];
double mean = 0.0;
int k = 0;
if (src.channels() == 3)
{
cvtColor(src, src, CV_BGR2GRAY);
img = Mat_<double>(src);
}
else
{
img = Mat_<double>(src);
}


/* 第一步,缩放尺寸*/
resize(img, img, Size(8, 8));


/* 第二步,离散余弦变换,DCT系数求取*/
dct(img, dst);


/* 第三步,求取DCT系数均值(左上角8*8区块的DCT系数)*/
for (int i = 0; i < 8; ++i) {
for (int j = 0; j < 8; ++j)
{
dIdex[k] = dst.at<double>(i, j);
mean += dst.at<double>(i, j) / 64;
++k;
}
}


/* 第四步,计算哈希值。*/
for (int i = 0; i<64; ++i)
{
if (dIdex[i] >= mean)
{
rst[i] = '1';
}
else
{
rst[i] = '0';
}
}
return rst;
}
//汉明距离计算  
int HanmingDistance(string &str1, string &str2)
{
if ((str1.size() != 64) || (str2.size() != 64))
return -1;
int difference = 0;
for (int i = 0; i<64; i++)
{
if (str1[i] != str2[i])
difference++;
}
return difference;
}




感知哈希(pHash)算法的缺点

1.算法得到的海明距离对于相同大小有旋转的图像距离值都较大。

2.若图像有遮挡或者缩放该算法均不适用。


2  BRISK特征提取算法

#include <cv.h>  
#include <opencv2/highgui/highgui.hpp>  
#include <opencv2/core/core.hpp>  
#include <opencv2/nonfree/features2d.hpp>  
#include <opencv2/nonfree/nonfree.hpp>  
#include <Windows.h>  
using namespace cv;
using namespace std;
int main()
{
//Load Image  
Mat c_src1 = imread("1-130(1).bmp");
Mat c_src2 = imread("3-130(1).bmp");
Mat src1 = imread("1-130(1).bmp", CV_LOAD_IMAGE_GRAYSCALE);
Mat src2 = imread("3-130(1).bmp", CV_LOAD_IMAGE_GRAYSCALE);
if (!src1.data || !src2.data)
{
cout << "Error reading images " << std::endl;
return -1;
}
//feature detect  
BRISK detector;
vector<KeyPoint> kp1, kp2;
double start = GetTickCount();
detector.detect(src1, kp1);
detector.detect(src2, kp2);
//cv::BRISK extractor;  
Mat des1, des2;//descriptor  
detector.compute(src1, kp1, des1);
detector.compute(src2, kp2, des2);
Mat res1, res2;
int drawmode = DrawMatchesFlags::DRAW_RICH_KEYPOINTS;
drawKeypoints(c_src1, kp1, res1, Scalar::all(-1), drawmode);//画出特征点  
drawKeypoints(c_src2, kp2, res2, Scalar::all(-1), drawmode);
cout << "size of description of Img1: " << kp1.size() << endl;
cout << "size of description of Img2: " << kp2.size() << endl;


BFMatcher matcher(NORM_HAMMING);
vector<DMatch> matches;
matcher.match(des1, des2, matches);
double end = GetTickCount();
cout << "耗时:" << (end - start) << "ms" << endl;
Mat img_match;
drawMatches(src1, kp1, src2, kp2, matches, img_match);
cout << "number of matched points: " << matches.size() << endl;
namedWindow("matches", WINDOW_NORMAL);
imshow("matches", img_match);
cvWaitKey(0);
cvDestroyAllWindows();
return 0;
}