0
0

SIFT算法详解(附有完整代码)

说明：本文旨在给出 SIFT 算法的具体实现，而在 SIFT 详解上只是做出简单介绍，在这里可以给大家推荐一篇好文：https://blog.csdn.net/zddblog/article/details/7521424；结合这篇文章和下文的具体代码实现，我相信你能很快掌握并运用 SIFT 算法，加油哦！！！如有疑问大家可以一起讨论学习！！！

一、SIFT算法简介

  1.     SIFT，即尺度不变特征变换（Scale-invariant feature transform，SIFT），是用于图像处理领域的一种描述。这种描述具有尺度不变性，可在图像中检测出关键点，是一种局部特征描述子。该方法于1999年由David Lowe 首先发表于计算机视觉国际会议（International Conference on Computer Vision，ICCV），2004年再次经David Lowe整理完善后发表于International journal of computer vision（IJCV）。截止2014年8月，该论文单篇被引次数达25000余次。

1、SIFT算法的特点

（1） SIFT特征是图像的局部特征，其对旋转、尺度缩放、亮度变化保持不变性，对视角变化、仿射变换、噪声也保持一定程度的稳定性；

（2）独特性（Distinctiveness）好，信息量丰富，适用于在海量特征数据库中进行快速、准确的匹配；

（3）多量性，即使少数的几个物体也可以产生大量的SIFT特征向量；

（4）高速性，经优化的SIFT匹配算法甚至可以达到实时的要求；

（5）可扩展性，可以很方便的与其他形式的特征向量进行联合。

2、SIFT算法可以解决的问题

  1.     目标的自身状态、场景所处的环境和成像器材的成像特性等因素影响图像配准/目标识别跟踪的性能。而SIFT算法在一定程度上可解决：

（1）目标的旋转、缩放、平移（RST）

（2）图像仿射/投影变换（视点viewpoint）

（3）光照影响（illumination）

（4）目标遮挡（occlusion）

（5）杂物场景（clutter）

（6）噪声

  1.     SIFT算法的实质是在不同的尺度空间上查找关键点(特征点)，并计算出关键点的方向。SIFT所查找到的关键点是一些十分突出，不会因光照，仿射变换和噪音等因素而变化的点，如角点、边缘点、暗区的亮点及亮区的暗点等。

二、SIFT算法分为如下四步

1. 尺度空间极值检测

  1.     搜索所有尺度上的图像位置。通过高斯微分函数来识别潜在的对于尺度和旋转不变的兴趣点。

2. 关键点定位

  1.     在每个候选的位置上，通过一个拟合精细的模型来确定位置和尺度。关键点的选择依据于它们的稳定程度。

3. 方向确定

  1.     基于图像局部的梯度方向，分配给每个关键点位置一个或多个方向。所有后面的对图像数据的操作都相对于关键点的方向、尺度和位置进行变换，从而提供对于这些变换的不变性。

4. 关键点描述

  1.     在每个关键点周围的邻域内，在选定的尺度上测量图像局部的梯度。这些梯度被变换成一种表示，这种表示允许比较大的局部形状的变形和光照变化。

三、SIFT具体实现

说明：代码实现主要有七个源文件，我和大家一样有时候下载了好几个资源然而没有一个能用的，导致最后都不知道要下载哪一个、是否要下载；所以这里直接把代码贴出来了！

1、mySift.h

  1. #pragma once                    //防止头文件重复包含和下面作用一样
  3. #include<iostream>
  4. #include<vector>
  5. #include<opencv2\core\core.hpp>
  6. #include<opencv2\features2d\features2d.hpp>
  8. using namespace std;
  9. using namespace cv;
  11. /*************************定义常量*****************************/
  13. //高斯核大小和标准差关系，size=2*(GAUSS_KERNEL_RATIO*sigma)+1,经常设置GAUSS_KERNEL_RATIO=2-3之间
  14. const double GAUSS_KERNEL_RATIO = 3;
  16. const int MAX_OCTAVES = 8;                    //金字塔最大组数
  18. const float CONTR_THR = 0.04f;                //默认是的对比度阈值(D(x))
  20. const float CURV_THR = 10.0f;                //关键点主曲率阈值
  22. const float INIT_SIGMA = 0.5f;                //输入图像的初始尺度
  24. const int IMG_BORDER = 2;                    //图像边界忽略的宽度，也可以改为 1 
  26. const int MAX_INTERP_STEPS = 5;                //关键点精确插值次数
  28. const int ORI_HIST_BINS = 36;                //计算特征点方向直方图的BINS个数
  30. const float ORI_SIG_FCTR = 1.5f;            //计算特征点主方向时候，高斯窗口的标准差因子
  32. const float ORI_RADIUS = 3 * ORI_SIG_FCTR;    //计算特征点主方向时，窗口半径因子
  34. const float ORI_PEAK_RATIO = 0.8f;            //计算特征点主方向时，直方图的峰值比
  36. const int DESCR_WIDTH = 4;                    //描述子直方图的网格大小(4x4)
  38. const int DESCR_HIST_BINS = 8;                //每个网格中直方图角度方向的维度
  40. const float DESCR_MAG_THR = 0.2f;            //描述子幅度阈值
  42. const float DESCR_SCL_FCTR = 3.0f;            //计算描述子时，每个网格的大小因子
  44. const int SAR_SIFT_GLOH_ANG_GRID = 8;        //GLOH网格沿角度方向等分区间个数
  46. const int SAR_SIFT_DES_ANG_BINS = 8;        //像素梯度方向在0-360度内等分区间个数
  48. const float SAR_SIFT_RADIUS_DES = 12.0f;    //描述子邻域半径        
  50. const int Mmax = 8;                            //像素梯度方向在0-360度内等分区间个数
  52. const double T = 100.0;                            //sobel算子去除冗余特征点的阈值
  54. const float SAR_SIFT_GLOH_RATIO_R1_R2 = 0.73f;//GLOH网格中间圆半径和外圆半径之比
  56. const float SAR_SIFT_GLOH_RATIO_R1_R3 = 0.25f;//GLOH网格最内层圆半径和外圆半径之比
  58. #define Feature_Point_Minimum 1500              //输入图像特征点最小个数
  60. #define We 0.2
  62. #define Wn 0.5
  64. #define Row_num 3
  66. #define Col_num 3
  68. #define SIFT_FIXPT_SCALE 48                    //不理解，后面可查原论文
  70. /************************sift类*******************************/
  71. class mySift
  72. {
  74. public:
  75.     //默认构造函数
  76.     mySift(int nfeatures = 0, int nOctaveLayers = 3, double contrastThreshold = 0.03,
  77.         double edgeThreshold = 10, double sigma = 1.6, bool double_size = true) :nfeatures(nfeatures),
  78.         nOctaveLayers(nOctaveLayers), contrastThreshold(contrastThreshold),
  79.         edgeThreshold(edgeThreshold), sigma(sigma), double_size(double_size) {}
  81.     //获得尺度空间每组中间层数
  82.     int get_nOctave_layers() { return nOctaveLayers; }
  84.     //获得图像尺度是否扩大一倍
  85.     bool get_double_size() { return double_size; }
  87.     //计算金字塔组数
  88.     int num_octaves(const Mat& image);
  90.     //生成高斯金字塔第一组，第一层图像
  91.     void create_initial_image(const Mat& image, Mat& init_image);
  93.     //使用 sobel 算子创建高斯金字塔第一层图像，以减少冗余特征点
  94.     void sobel_create_initial_image(const Mat& image, Mat& init_image);
  96.     //创建高斯金字塔
  97.     void build_gaussian_pyramid(const Mat& init_image, vector<vector<Mat>>& gauss_pyramid, int nOctaves);
  99.     //创建高斯差分金字塔
  100.     void build_dog_pyramid(vector<vector<Mat>>& dog_pyramid, const vector<vector<Mat>>& gauss_pyramid);
  102.     //该函数生成高斯差分金字塔
  103.     void amplit_orient(const Mat& image, vector<Mat>& amplit, vector<Mat>& orient, float scale, int nums);
  105.     //DOG金字塔特征点检测
  106.     void find_scale_space_extrema(const vector<vector<Mat>>& dog_pyr, const vector<vector<Mat>>& gauss_pyr,
  107.         vector<KeyPoint>& keypoints);
  109.     //DOG金字塔特征点检测，特征点方向细化版
  110.     void find_scale_space_extrema1(const vector<vector<Mat>>& dog_pyr, vector<vector<Mat>>& gauss_pyr,
  111.         vector<KeyPoint>& keypoints);
  113.     //计算特征点的描述子
  114.     void calc_sift_descriptors(const vector<vector<Mat>>& gauss_pyr, vector<KeyPoint>& keypoints,
  115.         Mat& descriptors, const vector<Mat>& amplit, const vector<Mat>& orient);
  117.     //构建空间尺度—主要是为了获取 amplit 和 orient 在使用 GLOH 描述子的时候使用
  118.     void build_sar_sift_space(const Mat& image, vector<Mat>& sar_harris_fun, vector<Mat>& amplit, vector<Mat>& orient);
  120.     //GLOH 计算一个特征描述子
  121.     void calc_gloh_descriptors(const vector<Mat>& amplit, const vector<Mat>& orient, const vector<KeyPoint>& keys, Mat& descriptors);
  123.     //特征点检测
  124.     void detect(const Mat& image, vector<vector<Mat>>& gauss_pyr, vector<vector<Mat>>& dog_pyr, vector<KeyPoint>& keypoints,
  125.         vector<vector<vector<float>>>& all_cell_contrasts,
  126.         vector<vector<float>>& average_contrast, vector<vector<int>>& n_cells, vector<int>& num_cell, vector<vector<int>>& available_n,
  127.         vector<int>& available_num, vector<KeyPoint>& final_keypoints,
  128.         vector<KeyPoint>& Final_keypoints, vector<KeyPoint>& Final_Keypoints);
  130.     //特征点描述
  131.     void comput_des(const vector<vector<Mat>>& gauss_pyr, vector<KeyPoint>& final_keypoints, const vector<Mat>& amplit,
  132.         const vector<Mat>& orient, Mat& descriptors);
  134. private:
  135.     int nfeatures;                //设定检测的特征点的个数值,如果此值设置为0，则不影响结果
  136.     int nOctaveLayers;            //每组金字塔中间层数
  137.     double contrastThreshold;    //对比度阈值（D(x)）
  138.     double edgeThreshold;        //特征点边缘曲率阈值
  139.     double sigma;                //高斯尺度空间初始层的尺度
  140.     bool double_size;            //是否上采样原始图像
  142. };//注意类结束的分号

2、mySift.cpp

  1. #include"mySift.h"
  3. #include<string>
  4. #include<cmath>
  5. #include<iostream>            //输入输出
  6. #include<vector>            //vector
  7. #include<algorithm>
  8. #include<numeric>            //用于容器元素求和
  10. #include<opencv2\core\hal\hal.hpp>
  11. #include<opencv2\core\hal\intrin.hpp>
  13. #include<opencv2\opencv.hpp>
  14. #include<opencv2\core\core.hpp>                //opencv基本数据结构
  15. #include<opencv2\highgui\highgui.hpp>        //图像界面
  16. #include<opencv2\imgproc\imgproc.hpp>        //基本图像处理函数
  17. #include<opencv2\features2d\features2d.hpp> //特征提取
  18. #include<opencv2\imgproc\types_c.h>
  19. #include<opencv2\videoio.hpp>
  20. #include <highgui\highgui.hpp>
  21. #include <imgproc\imgproc.hpp>
  23. /******************根据输入图像大小计算高斯金字塔的组数****************************/
  24. /*image表示原始输入灰度图像,inline函数必须在声明处定义
  25. double_size_image表示是否在构建金字塔之前上采样原始图像
  26. */
  27. int mySift::num_octaves(const Mat& image)
  28. {
  29.     int temp;
  30.     float size_temp = (float)min(image.rows, image.cols);
  31.     temp = cvRound(log(size_temp) / log((float)2) - 2);
  33.     if (double_size)
  34.         temp += 1;
  35.     if (temp > MAX_OCTAVES)                            //尺度空间最大组数设置为MAX_OCTAVES
  36.         temp = MAX_OCTAVES;
  38.     return temp;
  39. }
  41. /************************sobel滤波器计算高斯尺度空间图像梯度大小*****************************/
  42. void sobelfilter(Mat& image, Mat& G)
  43. {
  44.     // image 是经过归一化后的数据 (0,1)
  45.     int rows = image.rows;
  46.     int cols = image.cols;
  48.     float dx = 0.0, dy = 0.0;
  50.     //cv::Mat Gx = cv::Mat::zeros(rows, cols, CV_32FC1);    //包含了图像像素在水平方向上的导数的近似值的图像
  51.     //cv::Mat Gy = cv::Mat::zeros(rows, cols, CV_32FC1);    //包含了图像像素在垂直方向上的导数的近似值的图像
  53.     G = Mat::zeros(rows, cols, CV_32FC1);                    //在每个像素点处的灰度大小由Gx和Gy共同决定
  55.     double v = 0.0, vx, vy;
  57.     //利用 sobel 算子求梯度幅度图像
  58.     for (int i = 0; i < rows; i++)
  59.     {
  60.         for (int j = 0; j < cols; j++)
  61.         {
  62.             v = 0.0;
  64.             if (i == 0 || j == 0 || i == rows - 1 || j == cols - 1)
  65.             {
  66.                 G.at<float>(i, j) = 0.0;
  67.             }
  68.             else
  69.             {
  70.                 float dx = image.at<float>(i - 1, j + 1) - image.at<float>(i - 1, j - 1)
  71.                     + 2 * image.at<float>(i, j + 1) - 2 * image.at<float>(i, j - 1)
  72.                     + image.at<float>(i + 1, j + 1) - image.at<float>(i + 1, j - 1);
  74.                 float dy = image.at<float>(i + 1, j - 1) - image.at<float>(i - 1, j - 1)
  75.                     + 2 * image.at<float>(i + 1, j) - 2 * image.at<float>(i - 1, j) +
  76.                     image.at<float>(i + 1, j + 1) - image.at<float>(i - 1, j + 1);
  78.                 v = abs(dx) + abs(dy);            //简化后 G = |Gx| + |Gy|
  80.                 //保证像素值在有效范围内
  81.                 v = fmax(v, 0);                    //返回浮点数中较大的一个
  82.                 v = fmin(v, 255);                //返回浮点数中较小的一个
  84.                 if (v > T)                        //T为阈值等于50
  85.                     G.at<float>(i, j) = (float)v;
  86.                 else
  87.                     G.at<float>(i, j) = 0.0;
  88.             }
  89.         }
  90.     }
  92.     //水平方向上的导数的近似值的图像
  93.     /*for (int i = 0; i < rows; i++)
  94.     {
  95.         for (int j = 0; j < cols; j++)
  96.         {
  97.             vx = 0;
  99.             if (i == 0 || j == 0 || i == rows - 1 || j == cols - 1)
  100.                 Gx.at<float>(i, j) = 0;
  101.             else
  102.             {
  103.                 dx = image.at<float>(i - 1, j + 1) - image.at<float>(i - 1, j - 1)
  104.                     + 2 * image.at<float>(i, j + 1) - 2 * image.at<float>(i, j - 1)
  105.                     + image.at<float>(i + 1, j + 1) - image.at<float>(i + 1, j - 1);
  107.                 vx = abs(dx);
  109.                 vx = fmax(vx, 0); vx = fmin(vx, 255);
  111.                 Gx.at<float>(i, j) = (float)vx;
  112.             }
  113.         }
  114.     }*/
  116.     //垂直方向上的导数的近似值的图像
  117.     /*for (int i = 0; i < rows; i++)
  118.     {
  119.         for (int j = 0; j < cols; j++)
  120.         {
  121.             vy = 0;
  123.             if (i == 0 || j == 0 || i == rows - 1 || j == cols - 1)
  124.                 Gy.at<float>(i, j) = 0;
  125.             else
  126.             {
  127.                 dy = image.at<float>(i + 1, j - 1) - image.at<float>(i - 1, j - 1)
  128.                     + 2 * image.at<float>(i + 1, j) - 2 * image.at<float>(i - 1, j) +
  129.                     image.at<float>(i + 1, j + 1) - image.at<float>(i - 1, j + 1);
  131.                 vy = abs(dy);
  133.                 vy = fmax(vy, 0); vx = fmin(vy, 255);
  135.                 Gy.at<float>(i, j) = (float)vy;
  136.             }
  137.         }
  138.     }*/
  140.     //cv::imshow("Gx", Gx);                        // horizontal
  141.     //cv::imshow("Gy", Gy);                        // vertical
  142.     //cv::imshow("G", G);                        // gradient
  143. }
  145. /*********该函数根据尺度和窗口半径生成ROEWA滤波模板************/
  146. /*size表示核半径，因此核宽度是2*size+1
  147.  scale表示指数权重参数
  148.  kernel表示生成的滤波核
  149.  */
  150. static void roewa_kernel(int size, float scale, Mat& kernel)
  151. {
  152.     kernel.create(2 * size + 1, 2 * size + 1, CV_32FC1);
  153.     for (int i = -size; i <= size; ++i)
  154.     {
  155.         float* ptr_k = kernel.ptr<float>(i + size);
  156.         for (int j = -size; j <= size; ++j)
  157.         {
  158.             ptr_k[j + size] = exp(-1.f * (abs(i) + abs(j)) / scale);
  159.         }
  160.     }
  161. }
  163. /************************创建高斯金字塔第一组，第一层图像************************************/
  164. /*image表示输入原始图像
  165.  init_image表示生成的高斯尺度空间的第一层图像
  166.  */
  167. void mySift::create_initial_image(const Mat& image, Mat& init_image)
  168. {
  169.     Mat gray_image;
  171.     if (image.channels() != 1)
  172.         cvtColor(image, gray_image, CV_RGB2GRAY);        //转换为灰度图像
  173.     else
  174.         gray_image = image.clone();
  176.     Mat floatImage;                                        //转换到0-1之间的浮点类型数据归一化，方便接下来的处理
  178.     //float_image=(float)gray_image*(1.0/255.0)
  179.     gray_image.convertTo(floatImage, CV_32FC1, 1.0 / 255.0, 0);
  181.     double sig_diff = 0;
  183.     if (double_size)
  184.     {
  185.         Mat temp_image;
  187.         //通过插值的方法改变图像尺寸的大小
  188.         resize(floatImage, temp_image, Size(2 * floatImage.cols, 2 * floatImage.rows), 0.0, 0.0, INTER_LINEAR);
  190.         //高斯平滑的标准差，值较大时平滑效果比较明显
  191.         sig_diff = sqrt(sigma * sigma - 4.0 * INIT_SIGMA * INIT_SIGMA);
  193.         //高斯滤波窗口大小选择很重要，这里选择(4*sig_diff_1+1)-(6*sig_diff+1)之间，且四舍五入
  194.         int kernel_width = 2 * cvRound(GAUSS_KERNEL_RATIO * sig_diff) + 1;
  196.         Size kernel_size(kernel_width, kernel_width);
  198.         //对图像进行平滑处理(高斯模糊)，即降低图像的分辨率,高斯模糊是实现尺度变换的唯一变换核，并其实唯一的线性核
  199.         GaussianBlur(temp_image, init_image, kernel_size, sig_diff, sig_diff);
  200.     }
  201.     else
  202.     {
  203.         sig_diff = sqrt(sigma * sigma - 1.0 * INIT_SIGMA * INIT_SIGMA);
  205.         //高斯滤波窗口大小选择很重要，这里选择(4*sig_diff_1+1)-(6*sig_diff+1)之间
  206.         int kernel_width = 2 * cvRound(GAUSS_KERNEL_RATIO * sig_diff) + 1;
  208.         Size kernel_size(kernel_width, kernel_width);
  210.         GaussianBlur(floatImage, init_image, kernel_size, sig_diff, sig_diff);
  211.     }
  212. }
  214. /************************使用 sobel 算子创建高斯金字塔第一组，第一层图像****************************/
  215. //目的是为了减少冗余特征点
  216. void mySift::sobel_create_initial_image(const Mat& image, Mat& init_image)
  217. {
  218.     Mat gray_image, gray_images;                        //gray_images用于存放经过sobel算子操作后的图像
  220.     if (image.channels() != 1)
  221.         cvtColor(image, gray_image, CV_RGB2GRAY);        //转换为灰度图像
  222.     else
  223.         gray_image = image.clone();
  225.     sobelfilter(gray_image, gray_images);
  227.     Mat floatImage;                                        //转换到0-1之间的浮点类型数据归一化，方便接下来的处理
  229.     //float_image=(float)gray_image*(1.0/255.0)
  230.     gray_images.convertTo(floatImage, CV_32FC1, 1.0 / 255.0, 0);
  232.     double sig_diff = 0;
  234.     if (double_size)
  235.     {
  236.         Mat temp_image;
  238.         //通过插值的方法改变图像尺寸的大小
  239.         resize(floatImage, temp_image, Size(2 * floatImage.cols, 2 * floatImage.rows), 0.0, 0.0, INTER_LINEAR);
  241.         //高斯平滑的标准差，值较大时平滑效果比较明显
  242.         sig_diff = sqrt(sigma * sigma - 4.0 * INIT_SIGMA * INIT_SIGMA);
  244.         //高斯滤波窗口大小选择很重要，这里选择(4*sig_diff_1+1)-(6*sig_diff+1)之间，且四舍五入
  245.         int kernel_width = 2 * cvRound(GAUSS_KERNEL_RATIO * sig_diff) + 1;
  247.         Size kernel_size(kernel_width, kernel_width);
  249.         //对图像进行平滑处理(高斯模糊)，即降低图像的分辨率,高斯模糊是实现尺度变换的唯一变换核，并其实唯一的线性核
  250.         GaussianBlur(temp_image, init_image, kernel_size, sig_diff, sig_diff);
  251.     }
  252.     else
  253.     {
  254.         sig_diff = sqrt(sigma * sigma - 1.0 * INIT_SIGMA * INIT_SIGMA);
  256.         //高斯滤波窗口大小选择很重要，这里选择(4*sig_diff_1+1)-(6*sig_diff+1)之间
  257.         int kernel_width = 2 * cvRound(GAUSS_KERNEL_RATIO * sig_diff) + 1;
  259.         Size kernel_size(kernel_width, kernel_width);
  261.         GaussianBlur(floatImage, init_image, kernel_size, sig_diff, sig_diff);
  262.     }
  263. }
  265.  /**************************生成高斯金字塔*****************************************/
  266.  /*init_image表示已经生成的高斯金字塔第一层图像
  267.   gauss_pyramid表示生成的高斯金字塔
  268.   nOctaves表示高斯金字塔的组数
  269.  */
  270. void mySift::build_gaussian_pyramid(const Mat& init_image, vector<vector<Mat>>& gauss_pyramid, int nOctaves)
  271. {
  272.     vector<double> sig;
  274.     sig.push_back(sigma);
  276.     double k = pow(2.0, 1.0 / nOctaveLayers);                //高斯金字塔每一层的系数 k
  278.     for (int i = 1; i < nOctaveLayers + 3; ++i)
  279.     {
  280.         double prev_sig = pow(k, i - 1) * sigma;                //每一个尺度层的尺度
  281.         double curr_sig = k * prev_sig;
  283.         //组内每层的尺度坐标计算公式
  284.         sig.push_back(sqrt(curr_sig * curr_sig - prev_sig * prev_sig));
  285.     }
  287.     gauss_pyramid.resize(nOctaves);
  289.     for (int i = 0; i < nOctaves; ++i)
  290.     {
  291.         gauss_pyramid[i].resize(nOctaveLayers + 3);
  292.     }
  294.     for (int i = 0; i < nOctaves; ++i)                        //对于每一组
  295.     {
  296.         for (int j = 0; j < nOctaveLayers + 3; ++j)            //对于组内的每一层
  297.         {
  298.             if (i == 0 && j == 0)                            //第一组，第一层
  299.                 gauss_pyramid[0][0] = init_image;
  301.             else if (j == 0)
  302.             {
  303.                 resize(gauss_pyramid[i - 1][3], gauss_pyramid[i][0],
  304.                     Size(gauss_pyramid[i - 1][3].cols / 2,
  305.                         gauss_pyramid[i - 1][3].rows / 2), 0, 0, INTER_LINEAR);
  306.             }
  307.             else
  308.             {
  309.                 //高斯滤波窗口大小选择很重要，这里选择(4*sig_diff_1+1)-(6*sig_diff+1)之间
  310.                 int kernel_width = 2 * cvRound(GAUSS_KERNEL_RATIO * sig[j]) + 1;
  311.                 Size kernel_size(kernel_width, kernel_width);
  313.                 GaussianBlur(gauss_pyramid[i][j - 1], gauss_pyramid[i][j], kernel_size, sig[j], sig[j]);
  314.             }
  315.         }
  316.     }
  317. }
  319. /*******************生成高斯差分金字塔，即LOG金字塔*************************/
  320. /*dog_pyramid表示DOG金字塔
  321.  gauss_pyramin表示高斯金字塔*/
  322. void mySift::build_dog_pyramid(vector<vector<Mat>>& dog_pyramid, const vector<vector<Mat>>& gauss_pyramid)
  323. {
  324.     vector<vector<Mat>>::size_type nOctaves = gauss_pyramid.size();
  326.     for (vector<vector<Mat>>::size_type i = 0; i < nOctaves; ++i)
  327.     {
  328.         //用于存放每一个梯度中的所有尺度层
  329.         vector<Mat> temp_vec;
  331.         for (auto j = 0; j < nOctaveLayers + 2; ++j)
  332.         {
  333.             Mat temp_img = gauss_pyramid[i][j + 1] - gauss_pyramid[i][j];
  335.             temp_vec.push_back(temp_img);
  336.         }
  337.         dog_pyramid.push_back(temp_vec);
  339.         temp_vec.clear();
  340.     }
  341. }
  343. /***********生成高斯差分金字塔当前层对应的梯度幅度图像和梯度方向图像***********/
  344. /*image为高斯差分金字塔当前层图像
  345.  *amplit为当前层梯度幅度图像
  346.  *orient为当前层梯度方向图像
  347.  *scale当前层尺度
  348.  *nums为相对底层的层数
  349.  */
  350. void mySift::amplit_orient(const Mat& image, vector<Mat>& amplit, vector<Mat>& orient, float scale, int nums)
  351. {
  352.     //分配内存
  353.     amplit.resize(Mmax * nOctaveLayers);
  354.     orient.resize(Mmax * nOctaveLayers);
  356.     int radius = cvRound(2 * scale);
  358.     Mat kernel;                                                //kernel(2 * radius + 1, 2 * radius + 1, CV_32FC1);
  360.     roewa_kernel(radius, scale, kernel);                    //返回滤波核，也即指数部分,存放在矩阵的右下角
  362.     //四个滤波模板生成
  363.     Mat W34 = Mat::zeros(2 * radius + 1, 2 * radius + 1, CV_32FC1);        //把kernel矩阵下半部分复制到对应部分
  364.     Mat W12 = Mat::zeros(2 * radius + 1, 2 * radius + 1, CV_32FC1);        //把kernel矩阵上半部分复制到对应部分
  365.     Mat W14 = Mat::zeros(2 * radius + 1, 2 * radius + 1, CV_32FC1);        //把kernel矩阵右半部分复制到对应部分
  366.     Mat W23 = Mat::zeros(2 * radius + 1, 2 * radius + 1, CV_32FC1);        //把kernel矩阵左半部分复制到对应部分
  368.     kernel(Range(radius + 1, 2 * radius + 1), Range::all()).copyTo(W34(Range(radius + 1, 2 * radius + 1), Range::all()));
  369.     kernel(Range(0, radius), Range::all()).copyTo(W12(Range(0, radius), Range::all()));
  370.     kernel(Range::all(), Range(radius + 1, 2 * radius + 1)).copyTo(W14(Range::all(), Range(radius + 1, 2 * radius + 1)));
  371.     kernel(Range::all(), Range(0, radius)).copyTo(W23(Range::all(), Range(0, radius)));
  373.     //滤波
  374.     Mat M34, M12, M14, M23;
  375.     double eps = 0.00001;
  377.     //float_image为图像归一化后的图像数据，做卷积运算
  378.     filter2D(image, M34, CV_32FC1, W34, Point(-1, -1), eps);
  379.     filter2D(image, M12, CV_32FC1, W12, Point(-1, -1), eps);
  380.     filter2D(image, M14, CV_32FC1, W14, Point(-1, -1), eps);
  381.     filter2D(image, M23, CV_32FC1, W23, Point(-1, -1), eps);
  383.     //计算水平梯度和竖直梯度
  384.     Mat Gx, Gy;
  385.     log((M14) / (M23), Gx);
  386.     log((M34) / (M12), Gy);
  388.     //计算梯度幅度和梯度方向
  389.     magnitude(Gx, Gy, amplit[nums]);                    //梯度幅度图像，平方和开平方
  390.     phase(Gx, Gy, orient[nums], true);                    //梯度方向图像
  392. }
  394. /***********************该函数计算尺度空间特征点的主方向，用于后面特征点的检测***************************/
  395. /*image表示特征点所在位置的高斯图像，后面可对着源码进行修改
  396.  pt表示特征点的位置坐标(x,y)
  397.  scale特征点的尺度
  398.  n表示直方图bin个数
  399.  hist表示计算得到的直方图
  400.  函数返回值是直方图hist中的最大数值*/
  401. static float clac_orientation_hist(const Mat& image, Point pt, float scale, int n, float* hist)
  402. {
  403.     int radius = cvRound(ORI_RADIUS * scale);            //特征点邻域半径(3*1.5*scale)
  405.     int len = (2 * radius + 1) * (2 * radius + 1);        //特征点邻域像素总个数（最大值）
  407.     float sigma = ORI_SIG_FCTR * scale;                    //特征点邻域高斯权重标准差(1.5*scale)
  409.     float exp_scale = -1.f / (2 * sigma * sigma);        //权重的指数部分
  411.     //使用AutoBuffer分配一段内存，这里多出4个空间的目的是为了方便后面平滑直方图的需要
  412.     AutoBuffer<float> buffer((4 * len) + n + 4);
  414.     //X保存水平差分，Y保存数值差分，Mag保存梯度幅度，Ori保存梯度角度，W保存高斯权重
  415.     float* X = buffer, * Y = buffer + len, * Mag = Y, * Ori = Y + len, * W = Ori + len;
  416.     float* temp_hist = W + len + 2;                        //临时保存直方图数据
  418.     for (int i = 0; i < n; ++i)
  419.         temp_hist[i] = 0.f;                                //数据清零
  421.     //计算邻域像素的水平差分和竖直差分
  422.     int k = 0;
  424.     for (int i = -radius; i < radius; ++i)
  425.     {
  426.         int y = pt.y + i;                                //邻域点的纵坐标
  428.         if (y <= 0 || y >= image.rows - 1)
  429.             continue;
  431.         for (int j = -radius; j < radius; ++j)
  432.         {
  433.             int x = pt.x + j;
  435.             if (x <= 0 || x >= image.cols - 1)
  436.                 continue;
  438.             float dx = image.at<float>(y, x + 1) - image.at<float>(y, x - 1);    //水平差分
  439.             float dy = image.at<float>(y + 1, x) - image.at<float>(y - 1, x);    //竖直差分
  441.             //保存水平差分和竖直差分及其对应的权重
  442.             X[k] = dx;
  443.             Y[k] = dy;
  444.             W[k] = (i * i + j * j) * exp_scale;
  446.             ++k;
  447.         }
  448.     }
  450.     len = k;                                                    //邻域内特征点的个数
  452.     cv::hal::exp(W, W, len);                                    //计算邻域内所有像素的高斯权重
  453.     cv::hal::fastAtan2(Y, X, Ori, len, true);                    //计算邻域内所有像素的梯度方向，角度范围0-360度
  454.     cv::hal::magnitude32f(X, Y, Mag, len);                        //计算邻域内所有像素的梯度幅度，计算的是数学意义上的梯度
  456.     //遍历邻域的像素
  457.     for (int i = 0; i < len; ++i)
  458.     {
  459.         int bin = cvRound((n / 360.f) * Ori[i]);                //利用像素的梯度方向，约束bin的范围在[0,(n-1)]
  461.         //像素点梯度方向为360度时，和0°一样
  462.         if (bin >= n)
  463.             bin = bin - n;
  465.         if (bin < 0)
  466.             bin = bin + n;
  468.         temp_hist[bin] = temp_hist[bin] + Mag[i] * W[i];        //统计邻域内像素各个方向在梯度直方图的幅值(加权后的幅值)
  469.     }
  471.     //平滑直方图
  472.     temp_hist[-1] = temp_hist[n - 1];
  473.     temp_hist[-2] = temp_hist[n - 2];
  474.     temp_hist[n] = temp_hist[0];
  475.     temp_hist[n + 1] = temp_hist[1];
  476.     for (int i = 0; i < n; ++i)
  477.     {
  478.         hist[i] = (temp_hist[i - 2] + temp_hist[i + 2]) * (1.f / 16.f) +
  479.             (temp_hist[i - 1] + temp_hist[i + 1]) * (4.f / 16.f) +
  480.             temp_hist[i] * (6.f / 16.f);
  481.     }
  483.     //获得直方图中最大值
  484.     float max_value = hist[0];
  485.     for (int i = 1; i < n; ++i)
  486.     {
  487.         if (hist[i] > max_value)
  488.             max_value = hist[i];
  489.     }
  490.     return max_value;
  491. }
  493. /***********************使用 sobel 滤波器定义的新梯度计算尺度空间特征点的主方向**************************/
  494. static float clac_orientation_hist_2(Mat& image, Point pt, float scale, int n, float* hist)
  495. {
  496.     Mat output_image;                                    //使用 sobel 滤波器计算的图像的梯度幅度图像
  498.     sobelfilter(image, output_image);                    //使用 sobel 滤波器求高斯差分图像的梯度幅度图像
  500.     int radius = cvRound(ORI_RADIUS * scale);            //特征点邻域半径(3*1.5*scale)
  502.     int len = (2 * radius + 1) * (2 * radius + 1);        //特征点邻域像素总个数（最大值）
  504.     float sigma = ORI_SIG_FCTR * scale;                    //特征点邻域高斯权重标准差(1.5*scale)
  506.     float exp_scale = -1.f / (2 * sigma * sigma);        //权重的指数部分
  508.     //使用AutoBuffer分配一段内存，这里多出4个空间的目的是为了方便后面平滑直方图的需要
  509.     AutoBuffer<float> buffer((4 * len) + n + 4);
  511.     //X保存水平差分，Y保存数值差分，Mag保存梯度幅度，Ori保存梯度角度，W保存高斯权重
  512.     float* X = buffer, * Y = buffer + len, * Mag = Y, * Ori = Y + len, * W = Ori + len;
  513.     float* temp_hist = W + len + 2;                        //临时保存直方图数据
  515.     for (int i = 0; i < n; ++i)
  516.         temp_hist[i] = 0.f;                                //数据清零
  518.     //计算邻域像素的水平差分和竖直差分
  519.     int k = 0;
  521.     for (int i = -radius; i < radius; ++i)
  522.     {
  523.         int y = pt.y + i;                                //邻域点的纵坐标，行
  525.         if (y <= 0 || y >= output_image.rows - 1)
  526.             continue;
  528.         for (int j = -radius; j < radius; ++j)
  529.         {
  530.             int x = pt.x + j;                            //邻域点的纵坐标，列
  532.             if (x <= 0 || x >= output_image.cols - 1)
  533.                 continue;
  535.             //float dx = image.at<float>(y, x + 1) - image.at<float>(y, x - 1);    //水平差分
  537.             float dx = output_image.at<float>(y - 1, x + 1) - output_image.at<float>(y - 1, x - 1)
  538.                 + 2 * output_image.at<float>(y, x + 1) - 2 * output_image.at<float>(y, x - 1)
  539.                 + output_image.at<float>(y + 1, x + 1) - output_image.at<float>(y + 1, x - 1);
  541.             float dy = output_image.at<float>(y + 1, x - 1) - output_image.at<float>(y - 1, x - 1)
  542.                 + 2 * output_image.at<float>(y + 1, x) - 2 * output_image.at<float>(y - 1, x) +
  543.                 output_image.at<float>(y + 1, x + 1) - output_image.at<float>(y - 1, x + 1);
  545.             /*float dx = image.at<float>(y - 1, x + 1) - image.at<float>(y - 1, x - 1)
  546.                 + 2 * image.at<float>(y, x + 1) - 2 * image.at<float>(y, x - 1)
  547.                 + image.at<float>(y + 1, x + 1) - image.at<float>(y + 1, x - 1);
  549.             float dy = image.at<float>(y + 1, x - 1) - image.at<float>(y - 1, x - 1)
  550.                 + 2 * image.at<float>(y + 1, x) - 2 * image.at<float>(y - 1, x) +
  551.                 image.at<float>(y + 1, x + 1) - image.at<float>(y - 1, x + 1);*/
  553.                 //保存水平差分和竖直差分及其对应的权重
  554.             X[k] = dx;
  555.             Y[k] = dy;
  556.             W[k] = (i * i + j * j) * exp_scale;
  558.             ++k;
  559.         }
  560.     }
  562.     len = k;                                                    //邻域内特征点的个数
  564.     cv::hal::exp(W, W, len);                                    //计算邻域内所有像素的高斯权重
  565.     cv::hal::fastAtan2(Y, X, Ori, len, true);                    //计算邻域内所有像素的梯度方向，角度范围0-360度
  566.     cv::hal::magnitude32f(X, Y, Mag, len);                        //计算邻域内所有像素的梯度幅度，计算的是数学意义上的梯度
  568.     //遍历邻域的像素
  569.     for (int i = 0; i < len; ++i)
  570.     {
  571.         int bin = cvRound((n / 360.f) * Ori[i]);                //利用像素的梯度方向，约束bin的范围在[0,(n-1)]
  573.         //像素点梯度方向为360度时，和0°一样
  574.         if (bin >= n)
  575.             bin = bin - n;
  577.         if (bin < 0)
  578.             bin = bin + n;
  580.         temp_hist[bin] = temp_hist[bin] + Mag[i] * W[i];        //统计邻域内像素各个方向在梯度直方图的幅值(加权后的幅值)
  581.     }
  583.     //平滑直方图
  584.     temp_hist[-1] = temp_hist[n - 1];
  585.     temp_hist[-2] = temp_hist[n - 2];
  586.     temp_hist[n] = temp_hist[0];
  587.     temp_hist[n + 1] = temp_hist[1];
  588.     for (int i = 0; i < n; ++i)
  589.     {
  590.         hist[i] = (temp_hist[i - 2] + temp_hist[i + 2]) * (1.f / 16.f) +
  591.             (temp_hist[i - 1] + temp_hist[i + 1]) * (4.f / 16.f) +
  592.             temp_hist[i] * (6.f / 16.f);
  593.     }
  595.     //获得直方图中最大值
  596.     float max_value = hist[0];
  597.     for (int i = 1; i < n; ++i)
  598.     {
  599.         if (hist[i] > max_value)
  600.             max_value = hist[i];
  601.     }
  602.     return max_value;
  603. }
  605. /******************该函数计算特征点主方向,用于LOGH版本*********************/
  606. /*amplit表示特征点所在层的梯度幅度，即输入图像对应像素点的梯度存在了对应位置
  607.   orient表示特征点所在层的梯度方向，0-360度
  608.   point表示特征点坐标
  609.   scale表示特征点的所在层的尺度
  610.   hist表示生成的直方图
  611.   n表示主方向直方图bin个数
  612.  */
  613. static float calc_orient_hist(const Mat& amplit, const Mat& orient, Point pt, float scale, float* hist, int n)
  614. {
  615.     //暂且认为是只进行了下采样，没有进行高斯模糊
  616.     int num_row = amplit.rows;
  617.     int num_col = amplit.cols;
  619.     Point point(cvRound(pt.x), cvRound(pt.y));
  621.     //int radius = cvRound(SAR_SIFT_FACT_RADIUS_ORI * scale);
  622.     int radius = cvRound(6 * scale);
  624.     radius = min(radius, min(num_row / 2, num_col / 2));
  626.     float gauss_sig = 2 * scale;                            //高斯加权标准差
  628.     float exp_temp = -1.f / (2 * gauss_sig * gauss_sig);    //权重指数部分
  630.     //邻域区域
  631.     int radius_x_left = point.x - radius;
  632.     int radius_x_right = point.x + radius;
  633.     int radius_y_up = point.y - radius;
  634.     int radius_y_down = point.y + radius;
  636.     //防止越界
  637.     if (radius_x_left < 0)
  638.         radius_x_left = 0;
  639.     if (radius_x_right > num_col - 1)
  640.         radius_x_right = num_col - 1;
  641.     if (radius_y_up < 0)
  642.         radius_y_up = 0;
  643.     if (radius_y_down > num_row - 1)
  644.         radius_y_down = num_row - 1;
  646.     //此时特征点周围矩形区域相对于本矩形区域的中心
  647.     int center_x = point.x - radius_x_left;
  648.     int center_y = point.y - radius_y_up;
  650.     //矩形区域的边界，计算高斯权值
  651.     Range x_rng(-(point.x - radius_x_left), radius_x_right - point.x);
  652.     Range y_rng(-(point.y - radius_y_up), radius_y_down - point.y);
  654.     Mat gauss_weight;
  656.     gauss_weight.create(y_rng.end - y_rng.start + 1, x_rng.end - x_rng.start + 1, CV_32FC1);
  658.     //求各个像素点的高斯权重
  659.     for (int i = y_rng.start; i <= y_rng.end; ++i)
  660.     {
  661.         float* ptr_gauss = gauss_weight.ptr<float>(i - y_rng.start);
  663.         for (int j = x_rng.start; j <= x_rng.end; ++j)
  664.             ptr_gauss[j - x_rng.start] = exp((i * i + j * j) * exp_temp);
  665.     }
  667.     //索引特征点周围的像素梯度幅度，梯度方向
  668.     Mat sub_amplit = amplit(Range(radius_y_up, radius_y_down + 1), Range(radius_x_left, radius_x_right + 1));
  669.     Mat sub_orient = orient(Range(radius_y_up, radius_y_down + 1), Range(radius_x_left, radius_x_right + 1));
  671.     //Mat W = sub_amplit.mul(gauss_weight);            //加入高斯权重，计算高斯权重时，正确匹配点对反而变少了
  672.     Mat W = sub_amplit;                                //没加高斯权重，梯度幅值
  674.     //计算直方图
  675.     AutoBuffer<float> buffer(n + 4);
  677.     float* temp_hist = buffer + 2;
  679.     for (int i = 0; i < n; ++i)
  680.         temp_hist[i] = 0.f;
  682.     for (int i = 0; i < sub_orient.rows; i++)
  683.     {
  684.         float* ptr_1 = W.ptr<float>(i);
  685.         float* ptr_2 = sub_orient.ptr<float>(i);
  687.         for (int j = 0; j < sub_orient.cols; j++)
  688.         {
  689.             if (((i - center_y) * (i - center_y) + (j - center_x) * (j - center_x)) < radius * radius)
  690.             {
  691.                 int bin = cvRound(ptr_2[j] * n / 360.f);
  693.                 if (bin > n)
  694.                     bin = bin - n;
  695.                 if (bin < 0)
  696.                     bin = bin + n;
  697.                 temp_hist[bin] += ptr_1[j];
  698.             }
  699.         }
  700.     }
  702.     //平滑直方图，可以防止突变
  703.     temp_hist[-1] = temp_hist[n - 1];
  704.     temp_hist[-2] = temp_hist[n - 2];
  705.     temp_hist[n] = temp_hist[0];
  706.     temp_hist[n + 1] = temp_hist[1];
  708.     for (int i = 0; i < n; ++i)
  709.     {
  710.         hist[i] = (temp_hist[i - 2] + temp_hist[i + 2]) * (1.f / 16.f) +
  711.             (temp_hist[i - 1] + temp_hist[i + 1]) * (4.f / 16.f) +
  712.             temp_hist[i] * (6.f / 16.f);
  713.     }
  715.     //获得直方图中最大值
  716.     float max_value = hist[0];
  717.     for (int i = 1; i < n; ++i)
  718.     {
  719.         if (hist[i] > max_value)
  720.             max_value = hist[i];
  721.     }
  722.     return max_value;
  723. }
  725. /****************************该函数精确定位特征点位置(x,y,scale)，用于后面特征点的检测*************************/
  726. /*功能：确定特征点的位置，并通过主曲率消除边缘相应点,该版本是简化版
  727.  dog_pry表示DOG金字塔
  728.  kpt表示精确定位后该特征点的信息
  729.  octave表示初始特征点所在的组
  730.  layer表示初始特征点所在的层
  731.  row表示初始特征点在图像中的行坐标
  732.  col表示初始特征点在图像中的列坐标
  733.  nOctaveLayers表示DOG金字塔每组中间层数，默认是3
  734.  contrastThreshold表示对比度阈值，默认是0.04
  735.  edgeThreshold表示边缘阈值，默认是10
  736.  sigma表示高斯尺度空间最底层图像尺度，默认是1.6*/
  737. static bool adjust_local_extrema_1(const vector<vector<Mat>>& dog_pyr, KeyPoint& kpt, int octave, int& layer, int& row,
  738.     int& col, int nOctaveLayers, float contrastThreshold, float edgeThreshold, float sigma)
  739. {
  740.     float xi = 0, xr = 0, xc = 0;
  741.     int i = 0;
  742.     for (; i < MAX_INTERP_STEPS; ++i)                    //最大迭代次数
  743.     {
  744.         const Mat& img = dog_pyr[octave][layer];        //当前层图像索引
  745.         const Mat& prev = dog_pyr[octave][layer - 1];    //之前层图像索引
  746.         const Mat& next = dog_pyr[octave][layer + 1];    //下一层图像索引
  748.         //特征点位置x方向，y方向,尺度方向的一阶偏导数
  749.         float dx = (img.at<float>(row, col + 1) - img.at<float>(row, col - 1)) * (1.f / 2.f);
  750.         float dy = (img.at<float>(row + 1, col) - img.at<float>(row - 1, col)) * (1.f / 2.f);
  751.         float dz = (next.at<float>(row, col) - prev.at<float>(row, col)) * (1.f / 2.f);
  753.         //计算特征点位置二阶偏导数
  754.         float v2 = img.at<float>(row, col);
  755.         float dxx = img.at<float>(row, col + 1) + img.at<float>(row, col - 1) - 2 * v2;
  756.         float dyy = img.at<float>(row + 1, col) + img.at<float>(row - 1, col) - 2 * v2;
  757.         float dzz = prev.at<float>(row, col) + next.at<float>(row, col) - 2 * v2;
  759.         //计算特征点周围混合二阶偏导数
  760.         float dxy = (img.at<float>(row + 1, col + 1) + img.at<float>(row - 1, col - 1) -
  761.             img.at<float>(row + 1, col - 1) - img.at<float>(row - 1, col + 1)) * (1.f / 4.f);
  762.         float dxz = (next.at<float>(row, col + 1) + prev.at<float>(row, col - 1) -
  763.             next.at<float>(row, col - 1) - prev.at<float>(row, col + 1)) * (1.f / 4.f);
  764.         float dyz = (next.at<float>(row + 1, col) + prev.at<float>(row - 1, col) -
  765.             next.at<float>(row - 1, col) - prev.at<float>(row + 1, col)) * (1.f / 4.f);
  767.         Matx33f H(dxx, dxy, dxz, dxy, dyy, dyz, dxz, dyz, dzz);
  769.         Vec3f dD(dx, dy, dz);
  771.         Vec3f X = H.solve(dD, DECOMP_SVD);
  773.         xc = -X[0];        //x方向偏移量
  774.         xr = -X[1];        //y方向偏移量
  775.         xi = -X[2];        //尺度方向偏移量
  777.         //如果三个方向偏移量都小于0.5，说明已经找到特征点准确位置
  778.         if (abs(xc) < 0.5f && abs(xr) < 0.5f && abs(xi) < 0.5f)
  779.             break;
  781.         //如果其中一个方向的偏移量过大，则删除该点
  782.         if (abs(xc) > (float)(INT_MAX / 3) ||
  783.             abs(xr) > (float)(INT_MAX / 3) ||
  784.             abs(xi) > (float)(INT_MAX / 3))
  785.             return false;
  787.         col = col + cvRound(xc);
  788.         row = row + cvRound(xr);
  789.         layer = layer + cvRound(xi);
  791.         //如果特征点定位在边界区域，同样也需要删除
  792.         if (layer<1 || layer>nOctaveLayers ||
  793.             col<IMG_BORDER || col>img.cols - IMG_BORDER ||
  794.             row<IMG_BORDER || row>img.rows - IMG_BORDER)
  795.             return false;
  796.     }
  798.     //如果i=MAX_INTERP_STEPS，说明循环结束也没有满足条件，即该特征点需要被删除
  799.     if (i >= MAX_INTERP_STEPS)
  800.         return false;
  802.     /**************************再次删除低响应点(对比度较低的点)********************************/
  803.     //再次计算特征点位置x方向，y方向,尺度方向的一阶偏导数
  804.     //高对比度的特征对图像的变形是稳定的
  805.     {
  806.         const Mat& img = dog_pyr[octave][layer];
  807.         const Mat& prev = dog_pyr[octave][layer - 1];
  808.         const Mat& next = dog_pyr[octave][layer + 1];
  810.         float dx = (img.at<float>(row, col + 1) - img.at<float>(row, col - 1)) * (1.f / 2.f);
  811.         float dy = (img.at<float>(row + 1, col) - img.at<float>(row - 1, col)) * (1.f / 2.f);
  812.         float dz = (next.at<float>(row, col) - prev.at<float>(row, col)) * (1.f / 2.f);
  813.         Matx31f dD(dx, dy, dz);
  814.         float t = dD.dot(Matx31f(xc, xr, xi));
  816.         float contr = img.at<float>(row, col) + t * 0.5f;    //特征点响应 |D(x~)| 即对比度
  818.         //Low建议contr阈值是0.03，但是RobHess等建议阈值为0.04/nOctaveLayers
  819.         if (abs(contr) < contrastThreshold / nOctaveLayers)    //阈值设为0.03时特征点数量过多
  820.             return false;
  822.         /******************************删除边缘响应比较强的点************************************/
  824.         //再次计算特征点位置二阶偏导数，获取特征点出的 Hessian 矩阵，主曲率通过 2X2 的 Hessian 矩阵求出
  825.         //一个定义不好的高斯差分算子的极值在横跨边缘的地方有较大的主曲率而在垂直边缘的方向有较小的主曲率
  826.         float v2 = img.at<float>(row, col);
  827.         float dxx = img.at<float>(row, col + 1) + img.at<float>(row, col - 1) - 2 * v2;
  828.         float dyy = img.at<float>(row + 1, col) + img.at<float>(row - 1, col) - 2 * v2;
  829.         float dxy = (img.at<float>(row + 1, col + 1) + img.at<float>(row - 1, col - 1) -
  830.             img.at<float>(row + 1, col - 1) - img.at<float>(row - 1, col + 1)) * (1.f / 4.f);
  831.         float det = dxx * dyy - dxy * dxy;
  832.         float trace = dxx + dyy;
  834.         //主曲率和阈值的对比判定
  835.         if (det < 0 || (trace * trace * edgeThreshold >= det * (edgeThreshold + 1) * (edgeThreshold + 1)))
  836.             return false;
  838.         /*********到目前为止该特征的满足上面所有要求，保存该特征点信息***********/
  839.         kpt.pt.x = ((float)col + xc) * (1 << octave);    //相对于最底层的图像的x坐标
  840.         kpt.pt.y = ((float)row + xr) * (1 << octave);    //相对于最底层图像的y坐标
  841.         kpt.octave = octave + (layer << 8);                //组号保存在低字节，层号保存在高字节
  842.         //相对于最底层图像的尺度
  843.         kpt.size = sigma * powf(2.f, (layer + xi) / nOctaveLayers) * (1 << octave);
  844.         kpt.response = abs(contr);                        //特征点响应值(对比度)
  846.         return true;
  847.     }
  849. }
  851. /****************************该函数精确定位特征点位置(x,y,scale)，用于后面特征点的检测*************************/
  852. //该版本是 SIFT 原版，检测得到的特征点数量更多
  853. static bool adjust_local_extrema_2(const vector<vector<Mat>>& dog_pyr, KeyPoint& kpt, int octave, int& layer, int& row,
  854.     int& col, int nOctaveLayers, float contrastThreshold, float edgeThreshold, float sigma)
  855. {
  856.     const float img_scale = 1.f / (255 * SIFT_FIXPT_SCALE);    //SIFT_FIXPT_SCALE=48
  857.     const float deriv_scale = img_scale * 0.5f;
  858.     const float second_deriv_scale = img_scale;
  859.     const float cross_deriv_scale = img_scale * 0.25f;
  861.     float xi = 0, xr = 0, xc = 0;
  862.     int i = 0;
  864.     for (; i < MAX_INTERP_STEPS; ++i)                          //最大迭代次数
  865.     {
  866.         const Mat& img = dog_pyr[octave][layer];              //当前层图像索引
  867.         const Mat& prev = dog_pyr[octave][layer - 1];          //之前层图像索引
  868.         const Mat& next = dog_pyr[octave][layer + 1];          //下一层图像索引
  870.         //计算一阶偏导数，通过临近点差分求得
  871.         float dx = (img.at<float>(row, col + 1) - img.at<float>(row, col - 1)) * deriv_scale;
  872.         float dy = (img.at<float>(row + 1, col) - img.at<float>(row - 1, col)) * deriv_scale;
  873.         float dz = (next.at<float>(row, col) - prev.at<float>(row, col)) * deriv_scale;
  875.         //计算特征点位置二阶偏导数
  876.         //float v2  = img.at<float>(row, col);
  877.         float v2 = (float)img.at<float>(row, col) * 2.f;
  878.         float dxx = (img.at<float>(row, col + 1) + img.at<float>(row, col - 1) - v2) * second_deriv_scale;
  879.         float dyy = (img.at<float>(row + 1, col) + img.at<float>(row - 1, col) - v2) * second_deriv_scale;
  880.         float dzz = (prev.at<float>(row, col) + next.at<float>(row, col) - v2) * second_deriv_scale;
  882.         //计算特征点周围混合二阶偏导数
  883.         float dxy = (img.at<float>(row + 1, col + 1) + img.at<float>(row - 1, col - 1) -
  884.             img.at<float>(row + 1, col - 1) - img.at<float>(row - 1, col + 1)) * cross_deriv_scale;
  885.         float dxz = (next.at<float>(row, col + 1) + prev.at<float>(row, col - 1) -
  886.             next.at<float>(row, col - 1) - prev.at<float>(row, col + 1)) * cross_deriv_scale;
  887.         float dyz = (next.at<float>(row + 1, col) + prev.at<float>(row - 1, col) -
  888.             next.at<float>(row - 1, col) - prev.at<float>(row + 1, col)) * cross_deriv_scale;
  890.         Matx33f H(dxx, dxy, dxz, dxy, dyy, dyz, dxz, dyz, dzz);
  892.         Vec3f dD(dx, dy, dz);
  894.         Vec3f X = H.solve(dD, DECOMP_SVD);
  896.         xc = -X[0];        //x方向偏移量
  897.         xr = -X[1];        //y方向偏移量
  898.         xi = -X[2];        //尺度方向偏移量
  900.         //如果三个方向偏移量都小于0.5，说明已经找到特征点准确位置
  901.         if (abs(xc) < 0.5f && abs(xr) < 0.5f && abs(xi) < 0.5f)
  902.             break;
  904.         //如果其中一个方向的偏移量过大，则删除该点
  905.         if (abs(xc) > (float)(INT_MAX / 3) ||
  906.             abs(xr) > (float)(INT_MAX / 3) ||
  907.             abs(xi) > (float)(INT_MAX / 3))
  908.             return false;
  910.         col = col + cvRound(xc);
  911.         row = row + cvRound(xr);
  912.         layer = layer + cvRound(xi);
  914.         //如果特征点定位在边界区域，同样也需要删除
  915.         if (layer<1 || layer>nOctaveLayers ||
  916.             col < IMG_BORDER || col >= img.cols - IMG_BORDER ||
  917.             row < IMG_BORDER || row >= img.rows - IMG_BORDER)
  918.             return false;
  919.     }
  920.     //如果i=MAX_INTERP_STEPS，说明循环结束也没有满足条件，即该特征点需要被删除
  921.     if (i >= MAX_INTERP_STEPS)
  922.         return false;
  924.     /**************************再次删除低响应点(对比度较低的点)********************************/
  925.     //再次计算特征点位置x方向，y方向,尺度方向的一阶偏导数
  926.     //高对比度的特征对图像的变形是稳定的
  928.     const Mat& img = dog_pyr[octave][layer];
  929.     const Mat& prev = dog_pyr[octave][layer - 1];
  930.     const Mat& next = dog_pyr[octave][layer + 1];
  932.     float dx = (img.at<float>(row, col + 1) - img.at<float>(row, col - 1)) * deriv_scale;
  933.     float dy = (img.at<float>(row + 1, col) - img.at<float>(row - 1, col)) * deriv_scale;
  934.     float dz = (next.at<float>(row, col) - prev.at<float>(row, col)) * deriv_scale;
  935.     Matx31f dD(dx, dy, dz);
  936.     float t = dD.dot(Matx31f(xc, xr, xi));
  938.     float contr = img.at<float>(row, col) + t * 0.5f;    //特征点响应 |D(x~)| 即对比度
  940.     //Low建议contr阈值是0.03，但是RobHess等建议阈值为0.04/nOctaveLayers
  941.     if (abs(contr) < contrastThreshold / nOctaveLayers)    //阈值设为0.03时特征点数量过多
  942.         return false;
  944.     /******************************删除边缘响应比较强的点************************************/
  946.     //再次计算特征点位置二阶偏导数，获取特征点出的 Hessian 矩阵，主曲率通过 2X2 的 Hessian 矩阵求出
  947.     //一个定义不好的高斯差分算子的极值在横跨边缘的地方有较大的主曲率而在垂直边缘的方向有较小的主曲率
  948.     float v2 = (float)img.at<float>(row, col) * 2.f;
  949.     float dxx = (img.at<float>(row, col + 1) + img.at<float>(row, col - 1) - v2) * second_deriv_scale;
  950.     float dyy = (img.at<float>(row + 1, col) + img.at<float>(row - 1, col) - v2) * second_deriv_scale;
  951.     float dxy = (img.at<float>(row + 1, col + 1) + img.at<float>(row - 1, col - 1) -
  952.         img.at<float>(row + 1, col - 1) - img.at<float>(row - 1, col + 1)) * cross_deriv_scale;
  954.     float det = dxx * dyy - dxy * dxy;
  955.     float trace = dxx + dyy;
  957.     //主曲率和阈值的对比判定
  958.     if (det <= 0 || (trace * trace * edgeThreshold >= det * (edgeThreshold + 1) * (edgeThreshold + 1)))
  959.         return false;
  961.     /*********保存该特征点信息***********/
  962.     kpt.pt.x = ((float)col + xc) * (1 << octave);        //高斯金字塔坐标根据组数扩大相应的倍数
  963.     kpt.pt.y = ((float)row + xr) * (1 << octave);
  965.     // SIFT 描述子
  966.     kpt.octave = octave + (layer << 8) + (cvRound((xi + 0.5) * 255) << 16);                //特征点被检测出时所在的金字塔组
  968.     kpt.size = sigma * powf(2.f, (layer + xi) / nOctaveLayers) * (1 << octave) * 2;        //关键点邻域直径
  969.     kpt.response = abs(contr);                            //特征点响应值(对比度)
  971.     return true;
  973. }
  975. /************该函数在DOG金字塔上进行特征点检测，特征点精确定位，删除低对比度点，删除边缘响应较大点**********/
  976. /*dog_pyr表示高斯金字塔            原始 SIFT 算子
  977.  gauss_pyr表示高斯金字塔
  978.  keypoints表示检测到的特征点*/
  979. void mySift::find_scale_space_extrema(const vector<vector<Mat>>& dog_pyr, const vector<vector<Mat>>& gauss_pyr,
  980.     vector<KeyPoint>& keypoints)
  981. {
  982.     int nOctaves = (int)dog_pyr.size();                                //子八度个数
  984.     //Low文章建议thresholdThreshold是0.03，Rob Hess等人使用0.04/nOctaveLayers作为阈值
  985.     float threshold = (float)(contrastThreshold / nOctaveLayers);
  986.     const int n = ORI_HIST_BINS;                                    //n=36
  987.     float hist[n];
  988.     KeyPoint kpt;
  990.     keypoints.clear();                                                //先清空keypoints
  992.     for (int i = 0; i < nOctaves; ++i)                                //对于每一组
  993.     {
  994.         for (int j = 1; j <= nOctaveLayers; ++j)                    //对于组内每一层
  995.         {
  996.             const Mat& curr_img = dog_pyr[i][j];                    //当前层
  997.             const Mat& prev_img = dog_pyr[i][j - 1];                //上一层
  998.             const Mat& next_img = dog_pyr[i][j + 1];                //下一层
  1000.             int num_row = curr_img.rows;
  1001.             int num_col = curr_img.cols;                            //获得当前组图像的大小
  1002.             size_t step = curr_img.step1();                            //一行元素所占字节数
  1004.             //遍历每一个尺度层中的有效像素，像素值
  1005.             for (int r = IMG_BORDER; r < num_row - IMG_BORDER; ++r)
  1006.             {
  1007.                 const float* curr_ptr = curr_img.ptr<float>(r);        //指向的是第 r 行的起点，返回的是 float 类型的像素值
  1008.                 const float* prev_ptr = prev_img.ptr<float>(r - 1);
  1009.                 const float* next_ptr = next_img.ptr<float>(r + 1);
  1011.                 for (int c = IMG_BORDER; c < num_col - IMG_BORDER; ++c)
  1012.                 {
  1013.                     float val = curr_ptr[c];                        //当前中心点响应值
  1015.                     //开始检测特征点
  1016.                     if (abs(val) > threshold &&
  1017.                         ((val > 0 && val >= curr_ptr[c - 1] && val >= curr_ptr[c + 1] &&
  1018.                             val >= curr_ptr[c - step - 1] && val >= curr_ptr[c - step] && val >= curr_ptr[c - step + 1] &&
  1019.                             val >= curr_ptr[c + step - 1] && val >= curr_ptr[c + step] && val >= curr_ptr[c + step + 1] &&
  1020.                             val >= prev_ptr[c] && val >= prev_ptr[c - 1] && val >= prev_ptr[c + 1] &&
  1021.                             val >= prev_ptr[c - step - 1] && val >= prev_ptr[c - step] && val >= prev_ptr[c - step + 1] &&
  1022.                             val >= prev_ptr[c + step - 1] && val >= prev_ptr[c + step] && val >= prev_ptr[c + step + 1] &&
  1023.                             val >= next_ptr[c] && val >= next_ptr[c - 1] && val >= next_ptr[c + 1] &&
  1024.                             val >= next_ptr[c - step - 1] && val >= next_ptr[c - step] && val >= next_ptr[c - step + 1] &&
  1025.                             val >= next_ptr[c + step - 1] && val >= next_ptr[c + step] && val >= next_ptr[c + step + 1]) ||
  1026.                             (val < 0 && val <= curr_ptr[c - 1] && val <= curr_ptr[c + 1] &&
  1027.                                 val <= curr_ptr[c - step - 1] && val <= curr_ptr[c - step] && val <= curr_ptr[c - step + 1] &&
  1028.                                 val <= curr_ptr[c + step - 1] && val <= curr_ptr[c + step] && val <= curr_ptr[c + step + 1] &&
  1029.                                 val <= prev_ptr[c] && val <= prev_ptr[c - 1] && val <= prev_ptr[c + 1] &&
  1030.                                 val <= prev_ptr[c - step - 1] && val <= prev_ptr[c - step] && val <= prev_ptr[c - step + 1] &&
  1031.                                 val <= prev_ptr[c + step - 1] && val <= prev_ptr[c + step] && val <= prev_ptr[c + step + 1] &&
  1032.                                 val <= next_ptr[c] && val <= next_ptr[c - 1] && val <= next_ptr[c + 1] &&
  1033.                                 val <= next_ptr[c - step - 1] && val <= next_ptr[c - step] && val <= next_ptr[c - step + 1] &&
  1034.                                 val <= next_ptr[c + step - 1] && val <= next_ptr[c + step] && val <= next_ptr[c + step + 1])))
  1035.                     {
  1036.                         //++numKeys;
  1037.                         //获得特征点初始行号，列号，组号，组内层号
  1038.                         int octave = i, layer = j, r1 = r, c1 = c;
  1040.                         if (!adjust_local_extrema_1(dog_pyr, kpt, octave, layer, r1, c1,
  1041.                             nOctaveLayers, (float)contrastThreshold,
  1042.                             (float)edgeThreshold, (float)sigma))
  1043.                         {
  1044.                             continue;                                    //如果该初始点不满足条件，则不保存改点
  1045.                         }
  1047.                         float scale = kpt.size / float(1 << octave);    //该特征点相对于本组的尺度
  1049.                         //max_hist值对应的方向为主方向
  1050.                         float max_hist = clac_orientation_hist(gauss_pyr[octave][layer], Point(c1, r1), scale, n, hist);
  1052.                         //大于mag_thr值对应的方向为辅助方向
  1053.                         float mag_thr = max_hist * ORI_PEAK_RATIO;        //主峰值 80% 的方向作为辅助方向
  1055.                         //遍历直方图中的 36 个bin
  1056.                         for (int i = 0; i < n; ++i)
  1057.                         {
  1058.                             int left = i > 0 ? i - 1 : n - 1;
  1059.                             int right = i < n - 1 ? i + 1 : 0;
  1061.                             //创建新的特征点，大于主峰值 80% 的方向，赋值给该特征点，作为一个新的特征点；即有多个特征点，位置、尺度相同，方向不同
  1062.                             if (hist[i] > hist[left] && hist[i] > hist[right] && hist[i] >= mag_thr)
  1063.                             {
  1064.                                 float bin = i + 0.5f * (hist[left] - hist[right]) / (hist[left] + hist[right] - 2 * hist[i]);
  1065.                                 bin = bin < 0 ? n + bin : bin >= n ? bin - n : bin;
  1067.                                 kpt.angle = (360.f / n) * bin;            //原始 SIFT 算子使用的特征点的主方向0-360度
  1068.                                 keypoints.push_back(kpt);                //保存该特征点
  1070.                             }
  1071.                         }
  1072.                     }
  1073.                 }
  1074.             }
  1075.         }
  1076.     }
  1078.     //cout << "初始满足要求特征点个数是: " << numKeys << endl;
  1079. }
  1081. /************该函数在DOG金字塔上进行特征点检测，特征点精确定位，删除低对比度点，删除边缘响应较大点**********/
  1082. //对特征点进行方向的细化 + 增加更多的主方向版本 ——此时细化是对最后要给关键点进行赋值时的细化
  1083. //还可以考虑直接对方向直方图进行细化
  1084. void mySift::find_scale_space_extrema1(const vector<vector<Mat>>& dog_pyr, vector<vector<Mat>>& gauss_pyr,
  1085.     vector<KeyPoint>& keypoints)
  1086. {
  1087.     int nOctaves = (int)dog_pyr.size();                                //子八度个数
  1089.     //Low文章建议thresholdThreshold是0.03，Rob Hess等人使用0.04/nOctaveLayers作为阈值
  1090.     float threshold = (float)(contrastThreshold / nOctaveLayers);
  1091.     const int n = ORI_HIST_BINS;                                    //n=36
  1092.     float hist[n];
  1093.     KeyPoint kpt;
  1095.     vector<Mat> amplit;            //存放高斯差分金字塔每一层的梯度幅度图像
  1096.     vector<Mat> orient;            //存放高斯差分金字塔每一层的梯度方向图像
  1098.     keypoints.clear();                                                //先清空keypoints
  1100.     for (int i = 0; i < nOctaves; ++i)                                //对于每一组
  1101.     {
  1102.         for (int j = 1; j <= nOctaveLayers; ++j)                    //对于组内每一层
  1103.         {
  1104.             const Mat& curr_img = dog_pyr[i][j];                    //当前层
  1105.             const Mat& prev_img = dog_pyr[i][j - 1];                //上一层
  1106.             const Mat& next_img = dog_pyr[i][j + 1];                //下一层
  1108.             int num_row = curr_img.rows;
  1109.             int num_col = curr_img.cols;                            //获得当前组图像的大小
  1110.             size_t step = curr_img.step1();                            //一行元素所占字节数
  1112.             //遍历每一个尺度层中的有效像素，像素值
  1113.             for (int r = IMG_BORDER; r < num_row - IMG_BORDER; ++r)
  1114.             {
  1115.                 const float* curr_ptr = curr_img.ptr<float>(r);        //指向的是第 r 行的起点，返回的是 float 类型的像素值
  1116.                 const float* prev_ptr = prev_img.ptr<float>(r);
  1117.                 const float* next_ptr = next_img.ptr<float>(r);
  1119.                 for (int c = IMG_BORDER; c < num_col - IMG_BORDER; ++c)
  1120.                 {
  1121.                     float val = curr_ptr[c];                        //当前中心点响应值
  1123.                     //开始检测特征点
  1124.                     if (abs(val) > threshold &&
  1125.                         ((val > 0 && val >= curr_ptr[c - 1] && val >= curr_ptr[c + 1] &&
  1126.                             val >= curr_ptr[c - step - 1] && val >= curr_ptr[c - step] && val >= curr_ptr[c - step + 1] &&
  1127.                             val >= curr_ptr[c + step - 1] && val >= curr_ptr[c + step] && val >= curr_ptr[c + step + 1] &&
  1128.                             val >= prev_ptr[c] && val >= prev_ptr[c - 1] && val >= prev_ptr[c + 1] &&
  1129.                             val >= prev_ptr[c - step - 1] && val >= prev_ptr[c - step] && val >= prev_ptr[c - step + 1] &&
  1130.                             val >= prev_ptr[c + step - 1] && val >= prev_ptr[c + step] && val >= prev_ptr[c + step + 1] &&
  1131.                             val >= next_ptr[c] && val >= next_ptr[c - 1] && val >= next_ptr[c + 1] &&
  1132.                             val >= next_ptr[c - step - 1] && val >= next_ptr[c - step] && val >= next_ptr[c - step + 1] &&
  1133.                             val >= next_ptr[c + step - 1] && val >= next_ptr[c + step] && val >= next_ptr[c + step + 1]) ||
  1134.                             (val < 0 && val <= curr_ptr[c - 1] && val <= curr_ptr[c + 1] &&
  1135.                                 val <= curr_ptr[c - step - 1] && val <= curr_ptr[c - step] && val <= curr_ptr[c - step + 1] &&
  1136.                                 val <= curr_ptr[c + step - 1] && val <= curr_ptr[c + step] && val <= curr_ptr[c + step + 1] &&
  1137.                                 val <= prev_ptr[c] && val <= prev_ptr[c - 1] && val <= prev_ptr[c + 1] &&
  1138.                                 val <= prev_ptr[c - step - 1] && val <= prev_ptr[c - step] && val <= prev_ptr[c - step + 1] &&
  1139.                                 val <= prev_ptr[c + step - 1] && val <= prev_ptr[c + step] && val <= prev_ptr[c + step + 1] &&
  1140.                                 val <= next_ptr[c] && val <= next_ptr[c - 1] && val <= next_ptr[c + 1] &&
  1141.                                 val <= next_ptr[c - step - 1] && val <= next_ptr[c - step] && val <= next_ptr[c - step + 1] &&
  1142.                                 val <= next_ptr[c + step - 1] && val <= next_ptr[c + step] && val <= next_ptr[c + step + 1])))
  1143.                     {
  1144.                         //++numKeys;
  1145.                         //获得特征点初始行号，列号，组号，组内层号
  1146.                         int octave = i, layer = j, r1 = r, c1 = c, nums = i * nOctaves + j;
  1148.                         if (!adjust_local_extrema_2(dog_pyr, kpt, octave, layer, r1, c1,
  1149.                             nOctaveLayers, (float)contrastThreshold,
  1150.                             (float)edgeThreshold, (float)sigma))
  1151.                         {
  1152.                             continue;                                    //如果该初始点不满足条件，则不保存改点
  1153.                         }
  1155.                         float scale = kpt.size / float(1 << octave);    //该特征点相对于本组的尺度
  1157.                         //计算梯度幅度和梯度方向
  1158.                         //amplit_orient(curr_img, amplit, orient, scale, nums);
  1160.                         //max_hist值对应的方向为主方向
  1161.                         float max_hist = clac_orientation_hist(gauss_pyr[octave][layer], Point(c1, r1), scale, n, hist);
  1162.                         //float max_hist = calc_orient_hist(amplit[nums], orient[nums], Point2f(c1, r1), scale, hist, n);
  1164.                         大于mag_thr值对应的方向为辅助方向
  1165.                         //float mag_thr = max_hist * ORI_PEAK_RATIO;    //主峰值 80% 的方向作为辅助方向
  1167.                         //增加更多的主方向，以增加特征点对梯度差异的鲁棒性
  1168.                         float sum = 0.0;                                //直方图对应的幅值之和
  1169.                         float mag_thr = 0.0;                            //判断是否为主方向的阈值
  1171.                         for (int i = 0; i < n; ++i)
  1172.                         {
  1173.                             sum += hist[i];
  1174.                         }
  1175.                         mag_thr = 0.5 * (1.0 / 36) * sum;
  1177.                         //遍历直方图中的 36 个bin
  1178.                         for (int i = 0; i < n; ++i)
  1179.                         {
  1180.                             int left = i > 0 ? i - 1 : n - 1;
  1181.                             int right = i < n - 1 ? i + 1 : 0;
  1183.                             //创建新的特征点，大于主峰值 80% 的方向，赋值给该特征点，作为一个新的特征点；即有多个特征点，位置、尺度相同，方向不同
  1184.                             if (hist[i] > hist[left] && hist[i] > hist[right] && hist[i] >= mag_thr)
  1185.                             {
  1186.                                 float bin = i + 0.5f * (hist[left] - hist[right]) / (hist[left] + hist[right] - 2 * hist[i]);
  1187.                                 bin = bin < 0 ? n + bin : bin >= n ? bin - n : bin;
  1189.                                 //修改的地方，特征点的主方向修改为了0-180度，相当于对方向做了一个细化
  1190.                                 float angle = (360.f / n) * bin;
  1191.                                 if (angle >= 1 && angle <= 180)
  1192.                                 {
  1193.                                     kpt.angle = angle;
  1194.                                 }
  1195.                                 else if (angle > 180 && angle < 360)
  1196.                                 {
  1197.                                     kpt.angle = 360 - angle;
  1198.                                 }
  1200.                                 //kpt.angle = (360.f / n) * bin;            //原始 SIFT 算子使用的特征点的主方向0-360度
  1201.                                 keypoints.push_back(kpt);                    //保存该特征点
  1203.                             }
  1204.                         }
  1205.                     }
  1206.                 }
  1207.             }
  1208.         }
  1209.     }
  1211.     //cout << "初始满足要求特征点个数是: " << numKeys << endl;
  1212. }
  1214. /*该函数生成matlab中的meshgrid函数*/
  1215. /*x_range表示x方向的范围
  1216. y_range表示y方向的范围
  1217. X表示生成的沿x轴变化的网格
  1218. Y表示生成沿y轴变换的网格
  1219. */
  1220. static void meshgrid(const Range& x_range, const Range& y_range, Mat& X, Mat& Y)
  1221. {
  1222.     int x_start = x_range.start, x_end = x_range.end;
  1223.     int y_start = y_range.start, y_end = y_range.end;
  1224.     int width = x_end - x_start + 1;
  1225.     int height = y_end - y_start + 1;
  1227.     X.create(height, width, CV_32FC1);
  1228.     Y.create(height, width, CV_32FC1);
  1230.     for (int i = y_start; i <= y_end; i++)
  1231.     {
  1232.         float* ptr_1 = X.ptr<float>(i - y_start);
  1233.         for (int j = x_start; j <= x_end; ++j)
  1234.             ptr_1[j - x_start] = j * 1.0f;
  1235.     }
  1237.     for (int i = y_start; i <= y_end; i++)
  1238.     {
  1239.         float* ptr_2 = Y.ptr<float>(i - y_start);
  1240.         for (int j = x_start; j <= x_end; ++j)
  1241.             ptr_2[j - x_start] = i * 1.0f;
  1242.     }
  1243. }
  1245. /******************************计算一个特征点的描述子***********************************/
  1246. /*gauss_image表示特征点所在的高斯层
  1247.   main_angle表示特征点的主方向，角度范围是0-360度
  1248.   pt表示特征点在高斯图像上的坐标，相对与本组，不是相对于最底层
  1249.   scale表示特征点所在层的尺度，相对于本组，不是相对于最底层
  1250.   d表示特征点邻域网格宽度
  1251.   n表示每个网格内像素梯度角度等分个数
  1252.   descriptor表示生成的特征点的描述子*/
  1253. static void calc_sift_descriptor(const Mat& gauss_image, float main_angle, Point2f pt,
  1254.     float scale, int d, int n, float* descriptor)
  1255. {
  1256.     Point ptxy(cvRound(pt.x), cvRound(pt.y));                    //坐标取整
  1258.     float cos_t = cosf(-main_angle * (float)(CV_PI / 180));        //把角度转化为弧度，计算主方向的余弦
  1259.     float sin_t = sinf(-main_angle * (float)(CV_PI / 180));        //把角度转化为弧度，计算主方向的正弦
  1261.     float bins_per_rad = n / 360.f;                                // n = 8 ，梯度直方图分为 8 个方向
  1263.     float exp_scale = -1.f / (d * d * 0.5f);                    //权重指数部分
  1265.     float hist_width = DESCR_SCL_FCTR * scale;                    //特征点邻域内子区域边长，子区域的边长
  1267.     int radius = cvRound(hist_width * (d + 1) * sqrt(2) * 0.5f);//特征点邻域半径(d+1)*(d+1)，四舍五入
  1269.     int rows = gauss_image.rows, cols = gauss_image.cols;        //当前高斯层行、列信息
  1271.     //特征点邻域半径
  1272.     radius = min(radius, (int)sqrt((double)rows * rows + cols * cols));
  1274.     cos_t = cos_t / hist_width;
  1275.     sin_t = sin_t / hist_width;
  1277.     int len = (2 * radius + 1) * (2 * radius + 1);                //邻域内总像素数，为后面动态分配内存使用
  1279.     int histlen = (d + 2) * (d + 2) * (n + 2);                    //值为 360 
  1281.     AutoBuffer<float> buf(6 * len + histlen);
  1283.     //X保存水平差分，Y保存竖直差分，Mag保存梯度幅度，Angle保存特征点方向, W保存高斯权重
  1284.     float* X = buf, * Y = buf + len, * Mag = Y, * Angle = Y + len, * W = Angle + len;
  1285.     float* RBin = W + len, * CBin = RBin + len, * hist = CBin + len;
  1287.     //首先清空直方图数据
  1288.     for (int i = 0; i < d + 2; ++i)                // i 对应 row
  1289.     {
  1290.         for (int j = 0; j < d + 2; ++j)            // j 对应 col
  1291.         {
  1292.             for (int k = 0; k < n + 2; ++k)
  1294.                 hist[(i * (d + 2) + j) * (n + 2) + k] = 0.f;
  1295.         }
  1296.     }
  1298.     //把邻域内的像素分配到相应子区域内，计算子区域内每个像素点的权重(子区域即 d*d 中每一个小方格)
  1299.     int k = 0;
  1301.     //实际上是在 4 x 4 的网格中找 16 个种子点，每个种子点都在子网格的正中心，
  1302.     //通过三线性插值对不同种子点间的像素点进行加权作用到不同的种子点上绘制方向直方图
  1304.     for (int i = -radius; i < radius; ++i)                        // i 对应 y 行坐标
  1305.     {
  1306.         for (int j = -radius; j < radius; ++j)                    // j 对应 x 列坐标
  1307.         {
  1308.             float c_rot = j * cos_t - i * sin_t;                //旋转后邻域内采样点的 x 坐标
  1309.             float r_rot = j * sin_t + i * cos_t;                //旋转后邻域内采样点的 y 坐标
  1311.             //旋转后 5 x 5 的网格中的所有像素点被分配到 4 x 4 的网格中
  1312.             float cbin = c_rot + d / 2 - 0.5f;                    //旋转后的采样点落在子区域的 x 坐标
  1313.             float rbin = r_rot + d / 2 - 0.5f;                    //旋转后的采样点落在子区域的 y 坐标
  1315.             int r = ptxy.y + i, c = ptxy.x + j;                    //ptxy是高斯金字塔中的坐标
  1317.             //这里rbin，cbin范围是(-1,d)
  1318.             if (rbin > -1 && rbin < d && cbin>-1 && cbin < d && r>0 && r < rows - 1 && c>0 && c < cols - 1)
  1319.             {
  1320.                 float dx = gauss_image.at<float>(r, c + 1) - gauss_image.at<float>(r, c - 1);
  1321.                 float dy = gauss_image.at<float>(r + 1, c) - gauss_image.at<float>(r - 1, c);
  1323.                 X[k] = dx;                                                //邻域内所有像素点的水平差分
  1324.                 Y[k] = dy;                                                //邻域内所有像素点的竖直差分
  1326.                 CBin[k] = cbin;                                            //邻域内所有采样点落在子区域的 x 坐标
  1327.                 RBin[k] = rbin;                                            //邻域内所有采样点落在子区域的 y 坐标
  1329.                 W[k] = (c_rot * c_rot + r_rot * r_rot) * exp_scale;        //高斯权值的指数部分
  1331.                 ++k;
  1332.             }
  1333.         }
  1334.     }
  1336.     //计算采样点落在子区域的像素梯度幅度，梯度角度，和高斯权值
  1337.     len = k;
  1339.     cv::hal::exp(W, W, len);                        //邻域内所有采样点落在子区域的像素的高斯权值
  1340.     cv::hal::fastAtan2(Y, X, Angle, len, true);        //邻域内所有采样点落在子区域的像素的梯度方向，角度范围是0-360度
  1341.     cv::hal::magnitude(X, Y, Mag, len);                //邻域内所有采样点落在子区域的像素的梯度幅度
  1343.     //实际上是在 4 x 4 的网格中找 16 个种子点，每个种子点都在子网格的正中心，
  1344.     //通过三线性插值对不同种子点间的像素点进行加权作用到不同的种子点上绘制方向直方图
  1346.     //计算每个特征点的特征描述子
  1347.     for (k = 0; k < len; ++k)
  1348.     {
  1349.         float rbin = RBin[k], cbin = CBin[k];        //子区域内像素点坐标，rbin,cbin范围是(-1,d)
  1351.         改进的地方，对方向进行了一个细化，也是为了增加对梯度差异的鲁棒性
  1352.         //if (Angle[k] > 180 && Angle[k] < 360)
  1353.         //    Angle[k] = 360 - Angle[k];
  1355.         //子区域内像素点处理后的方向
  1356.         float temp = Angle[k] - main_angle;
  1358.         /*if (temp > 180 && temp < 360)
  1359.             temp = 360 - temp;*/
  1361.         float obin = temp * bins_per_rad;            //指定方向的数量后，邻域内像素点对应的方向
  1363.         float mag = Mag[k] * W[k];                    //子区域内像素点乘以权值后的梯度幅值
  1365.         int r0 = cvFloor(rbin);                        //ro取值集合是{-1，0，1，2，3}，没太懂为什么？
  1366.         int c0 = cvFloor(cbin);                        //c0取值集合是{-1，0，1，2，3}
  1367.         int o0 = cvFloor(obin);
  1369.         rbin = rbin - r0;                            //子区域内像素点坐标的小数部分，用于线性插值，分配像素点的作用
  1370.         cbin = cbin - c0;
  1371.         obin = obin - o0;                            //子区域方向的小数部分
  1373.         //限制范围为梯度直方图横坐标[0,n)，8 个方向直方图
  1374.         if (o0 < 0)
  1375.             o0 = o0 + n;
  1376.         if (o0 >= n)
  1377.             o0 = o0 - n;
  1379.         //三线性插值用于计算落在两个子区域之间的像素对两个子区域的作用，并把其分配到对应子区域的8个方向上
  1380.         //像素对应的信息通过加权分配给其周围的种子点，并把相应方向的梯度值进行累加  
  1382.         float v_r1 = mag * rbin;                    //第二行分配的值
  1383.         float v_r0 = mag - v_r1;                    //第一行分配的值
  1385.         float v_rc11 = v_r1 * cbin;                    //第二行第二列分配的值，右下角种子点
  1386.         float v_rc10 = v_r1 - v_rc11;                //第二行第一列分配的值，左下角种子点
  1388.         float v_rc01 = v_r0 * cbin;                    //第一行第二列分配的值，右上角种子点
  1389.         float v_rc00 = v_r0 - v_rc01;                //第一行第一列分配的值，左上角种子点
  1391.         //一个像素点的方向为每个种子点的两个方向做出贡献
  1392.         float v_rco111 = v_rc11 * obin;                //右下角种子点第二个方向上分配的值
  1393.         float v_rco110 = v_rc11 - v_rco111;            //右下角种子点第一个方向上分配的值
  1395.         float v_rco101 = v_rc10 * obin;
  1396.         float v_rco100 = v_rc10 - v_rco101;
  1398.         float v_rco011 = v_rc01 * obin;
  1399.         float v_rco010 = v_rc01 - v_rco011;
  1401.         float v_rco001 = v_rc00 * obin;
  1402.         float v_rco000 = v_rc00 - v_rco001;
  1404.         //该像素所在网格的索引
  1405.         int idx = ((r0 + 1) * (d + 2) + c0 + 1) * (n + 2) + o0;
  1407.         hist[idx] += v_rco000;
  1408.         hist[idx + 1] += v_rco001;
  1409.         hist[idx + n + 2] += v_rco010;
  1410.         hist[idx + n + 3] += v_rco011;
  1411.         hist[idx + (d + 2) * (n + 2)] += v_rco100;
  1412.         hist[idx + (d + 2) * (n + 2) + 1] += v_rco101;
  1413.         hist[idx + (d + 3) * (n + 2)] += v_rco110;
  1414.         hist[idx + (d + 3) * (n + 2) + 1] += v_rco111;
  1415.     }
  1417.     //由于圆周循环的特性，对计算以后幅角小于 0 度或大于 360 度的值重新进行调整，使
  1418.     //其在 0～360 度之间
  1419.     for (int i = 0; i < d; ++i)
  1420.     {
  1421.         for (int j = 0; j < d; ++j)
  1422.         {
  1423.             //类似于 hist[0][2][3] 第 0 行，第 2 列，种子点直方图中的第 3 个 bin
  1424.             int idx = ((i + 1) * (d + 2) + (j + 1)) * (n + 2);
  1426.             hist[idx] += hist[idx + n];
  1427.             //hist[idx + 1] += hist[idx + n + 1];//opencv源码中这句话是多余的,hist[idx + n + 1]永远是0.0
  1429.             for (k = 0; k < n; ++k)
  1430.                 descriptor[(i * d + j) * n + k] = hist[idx + k];
  1431.         }
  1432.     }
  1434.     //对描述子进行归一化
  1435.     int lenght = d * d * n;
  1436.     float norm = 0;
  1438.     //计算特征描述向量的模值的平方
  1439.     for (int i = 0; i < lenght; ++i)
  1440.     {
  1441.         norm = norm + descriptor[i] * descriptor[i];
  1442.     }
  1443.     norm = sqrt(norm);                            //特征描述向量的模值
  1445.     //此次归一化能去除光照的影响
  1446.     for (int i = 0; i < lenght; ++i)
  1447.     {
  1448.         descriptor[i] = descriptor[i] / norm;
  1449.     }
  1451.     //阈值截断,去除特征描述向量中大于 0.2 的值，能消除非线性光照的影响(相机饱和度对某些放的梯度影响较大，对方向的影响较小)
  1452.     for (int i = 0; i < lenght; ++i)
  1453.     {
  1454.         descriptor[i] = min(descriptor[i], DESCR_MAG_THR);
  1455.     }
  1457.     //再次归一化，能够提高特征的独特性
  1458.     norm = 0;
  1459.     for (int i = 0; i < lenght; ++i)
  1460.     {
  1461.         norm = norm + descriptor[i] * descriptor[i];
  1462.     }
  1463.     norm = sqrt(norm);
  1464.     for (int i = 0; i < lenght; ++i)
  1465.     {
  1466.         descriptor[i] = descriptor[i] / norm;
  1467.     }
  1468. }
  1470. /*************************该函数计算每个特征点的特征描述子*****************************/
  1471. /*amplit表示特征点所在层的梯度幅度图像
  1472.   orient表示特征点所在层的梯度角度图像
  1473.   pt表示特征点的位置
  1474.   scale表示特征点所在层的尺度
  1475.   main_ori表示特征点的主方向，0-360度
  1476.   d表示GLOH角度方向区间个数，默认是8，
  1477.   n表示每个网格内角度在0-360度之间等分个数，n默认是8
  1478.  */
  1479. static void calc_gloh_descriptor(const Mat& amplit, const Mat& orient, Point2f pt, float scale, float main_ori, int d, int n, float* ptr_des)
  1480. {
  1481.     Point point(cvRound(pt.x), cvRound(pt.y));
  1483.     //特征点旋转方向余弦和正弦
  1484.     float cos_t = cosf(-main_ori / 180.f * (float)CV_PI);
  1485.     float sin_t = sinf(-main_ori / 180.f * (float)CV_PI);
  1487.     int num_rows = amplit.rows;
  1488.     int num_cols = amplit.cols;
  1489.     int radius = cvRound(SAR_SIFT_RADIUS_DES * scale);
  1490.     radius = min(radius, min(num_rows / 2, num_cols / 2));//特征点邻域半径
  1492.     int radius_x_left = point.x - radius;
  1493.     int radius_x_right = point.x + radius;
  1494.     int radius_y_up = point.y - radius;
  1495.     int radius_y_down = point.y + radius;
  1497.     //防止越界
  1498.     if (radius_x_left < 0)
  1499.         radius_x_left = 0;
  1500.     if (radius_x_right > num_cols - 1)
  1501.         radius_x_right = num_cols - 1;
  1502.     if (radius_y_up < 0)
  1503.         radius_y_up = 0;
  1504.     if (radius_y_down > num_rows - 1)
  1505.         radius_y_down = num_rows - 1;
  1507.     //此时特征点周围本矩形区域的中心，相对于该矩形
  1508.     int center_x = point.x - radius_x_left;
  1509.     int center_y = point.y - radius_y_up;
  1511.     //特征点周围区域内像素梯度幅度，梯度角度
  1512.     Mat sub_amplit = amplit(Range(radius_y_up, radius_y_down + 1), Range(radius_x_left, radius_x_right + 1));
  1513.     Mat sub_orient = orient(Range(radius_y_up, radius_y_down + 1), Range(radius_x_left, radius_x_right + 1));
  1515.     //以center_x和center_y位中心，对下面矩形区域进行旋转
  1516.     Range x_rng(-(point.x - radius_x_left), radius_x_right - point.x);
  1517.     Range y_rng(-(point.y - radius_y_up), radius_y_down - point.y);
  1518.     Mat X, Y;
  1519.     meshgrid(x_rng, y_rng, X, Y);
  1520.     Mat c_rot = X * cos_t - Y * sin_t;
  1521.     Mat r_rot = X * sin_t + Y * cos_t;
  1522.     Mat GLOH_angle, GLOH_amplit;
  1523.     phase(c_rot, r_rot, GLOH_angle, true);//角度在0-360度之间
  1524.     GLOH_amplit = c_rot.mul(c_rot) + r_rot.mul(r_rot);//为了加快速度，没有计算开方
  1526.     //三个圆半径平方
  1527.     float R1_pow = (float)radius * radius;//外圆半径平方
  1528.     float R2_pow = pow(radius * SAR_SIFT_GLOH_RATIO_R1_R2, 2.f);//中间圆半径平方
  1529.     float R3_pow = pow(radius * SAR_SIFT_GLOH_RATIO_R1_R3, 2.f);//内圆半径平方
  1531.     int sub_rows = sub_amplit.rows;
  1532.     int sub_cols = sub_amplit.cols;
  1534.     //开始构建描述子,在角度方向对描述子进行插值
  1535.     int len = (d * 2 + 1) * n;
  1536.     AutoBuffer<float> hist(len);
  1537.     for (int i = 0; i < len; ++i)//清零
  1538.         hist[i] = 0;
  1540.     for (int i = 0; i < sub_rows; ++i)
  1541.     {
  1542.         float* ptr_amplit = sub_amplit.ptr<float>(i);
  1543.         float* ptr_orient = sub_orient.ptr<float>(i);
  1544.         float* ptr_GLOH_amp = GLOH_amplit.ptr<float>(i);
  1545.         float* ptr_GLOH_ang = GLOH_angle.ptr<float>(i);
  1546.         for (int j = 0; j < sub_cols; ++j)
  1547.         {
  1548.             if (((i - center_y) * (i - center_y) + (j - center_x) * (j - center_x)) < radius * radius)
  1549.             {
  1550.                 float pix_amplit = ptr_amplit[j];//该像素的梯度幅度
  1551.                 float pix_orient = ptr_orient[j];//该像素的梯度方向
  1552.                 float pix_GLOH_amp = ptr_GLOH_amp[j];//该像素在GLOH网格中的半径位置
  1553.                 float pix_GLOH_ang = ptr_GLOH_ang[j];//该像素在GLOH网格中的位置方向
  1555.                 int rbin, cbin, obin;
  1556.                 rbin = pix_GLOH_amp < R3_pow ? 0 : (pix_GLOH_amp > R2_pow ? 2 : 1);//rbin={0,1,2}
  1557.                 cbin = cvRound(pix_GLOH_ang * d / 360.f);
  1558.                 cbin = cbin > d ? cbin - d : (cbin <= 0 ? cbin + d : cbin);//cbin=[1,d]
  1560.                 obin = cvRound(pix_orient * n / 360.f);
  1561.                 obin = obin > n ? obin - n : (obin <= 0 ? obin + n : obin);//obin=[1,n]
  1563.                 if (rbin == 0)//内圆
  1564.                     hist[obin - 1] += pix_amplit;
  1565.                 else
  1566.                 {
  1567.                     int idx = ((rbin - 1) * d + cbin - 1) * n + n + obin - 1;
  1568.                     hist[idx] += pix_amplit;
  1569.                 }
  1570.             }
  1571.         }
  1572.     }
  1574.     //对描述子进行归一化
  1575.     float norm = 0;
  1576.     for (int i = 0; i < len; ++i)
  1577.     {
  1578.         norm = norm + hist[i] * hist[i];
  1579.     }
  1580.     norm = sqrt(norm);
  1581.     for (int i = 0; i < len; ++i)
  1582.     {
  1583.         hist[i] = hist[i] / norm;
  1584.     }
  1586.     //阈值截断
  1587.     for (int i = 0; i < len; ++i)
  1588.     {
  1589.         hist[i] = min(hist[i], DESCR_MAG_THR);
  1590.     }
  1592.     //再次归一化
  1593.     norm = 0;
  1594.     for (int i = 0; i < len; ++i)
  1595.     {
  1596.         norm = norm + hist[i] * hist[i];
  1597.     }
  1598.     norm = sqrt(norm);
  1599.     for (int i = 0; i < len; ++i)
  1600.     {
  1601.         ptr_des[i] = hist[i] / norm;
  1602.     }
  1604. }
  1606. /******************************计算一个特征点的描述子—改进版***********************************/
  1607. static void improve_calc_sift_descriptor(const Mat& gauss_image, float main_angle, Point2f pt,
  1608.     float scale, int d, int n, float* descriptor)
  1609. {
  1610.     int n1 = 16, n2 = 6, n3 = 4;
  1612.     Point ptxy(cvRound(pt.x), cvRound(pt.y));                    //坐标取整
  1614.     float cos_t = cosf(-main_angle * (float)(CV_PI / 180));        //计算主方向的余弦
  1615.     float sin_t = sinf(-main_angle * (float)(CV_PI / 180));        //计算主方向的正弦
  1617.     float bin_per_rad_1 = n1 / 360.f;                            //n=8
  1618.     float bin_per_rad_2 = n2 / 360.f;                            //原理特征点部分阈值
  1619.     float bin_per_rad_3 = n3 / 360.f;                            //原理特征点部分阈值
  1621.     float exp_scale = -1.f / (d * d * 0.5f);                    //权重指数部分
  1622.     float hist_width = DESCR_SCL_FCTR * scale;                    //子区域边长，子区域的面积也即采样像素点个数
  1624.     int radius = cvRound(hist_width * (d + 1) * sqrt(2) * 0.5f);//特征点邻域半径(d+1)*(d+1)
  1626.     int rows = gauss_image.rows, cols = gauss_image.cols;
  1628.     //特征点邻域半径
  1629.     radius = min(radius, (int)sqrt((double)rows * rows + cols * cols));
  1631.     cos_t = cos_t / hist_width;
  1632.     sin_t = sin_t / hist_width;
  1634.     int len = (2 * radius + 1) * (2 * radius + 1);                //邻域内总像素数
  1635.     int histlen = (d + 2) * (d + 2) * (n1 + 2);
  1637.     AutoBuffer<float> buf(6 * len + histlen);
  1639.     //X保存水平差分，Y保存竖直差分，Mag保存梯度幅度，Angle保存特征点方向, W保存高斯权重
  1640.     float* X = buf, * Y = buf + len, * Mag = Y, * Angle = Y + len, * W = Angle + len;
  1641.     float* X2 = buf, * Y2 = buf + len, * Mag2 = Y, * Angle2 = Y + len, * W2 = Angle + len;
  1642.     float* X3 = buf, * Y3 = buf + len, * Mag3 = Y, * Angle3 = Y + len, * W3 = Angle + len;
  1644.     float* RBin = W + len, * CBin = RBin + len, * hist = CBin + len;
  1645.     float* RBin2 = W + len, * CBin2 = RBin + len;
  1646.     float* RBin3 = W + len, * CBin3 = RBin + len;
  1648.     //首先清空直方图数据
  1649.     for (int i = 0; i < d + 2; ++i)
  1650.     {
  1651.         for (int j = 0; j < d + 2; ++j)
  1652.         {
  1653.             for (int k = 0; k < n + 2; ++k)
  1654.                 hist[(i * (d + 2) + j) * (n + 2) + k] = 0.f;
  1655.         }
  1656.     }
  1658.     //把邻域内的像素分配到相应子区域内，计算子区域内每个像素点的权重(子区域即 d*d 中每一个小方格)
  1660.     int k1 = 0, k2 = 0, k3 = 0;
  1662.     vector<int> v;                                    //存放外邻域像素点对应的序号
  1664.     for (int i = -radius; i < radius; ++i)
  1665.     {
  1666.         for (int j = -radius; j < radius; ++j)
  1667.         {
  1668.             float c_rot = j * cos_t - i * sin_t;    //旋转后邻域内采样点的 x 坐标
  1669.             float r_rot = j * sin_t + i * cos_t;    //旋转后邻域内采样点的 y 坐标
  1671.             float rbin = r_rot + d / 2 - 0.5f;        //旋转后的采样点落在子区域的 y 坐标
  1672.             float cbin = c_rot + d / 2 - 0.5f;        //旋转后的采样点落在子区域的 x 坐标
  1674.             int r = ptxy.y + i, c = ptxy.x + j;        //ptxy是高斯金字塔中的坐标
  1676.             //对离中心点近的部分进行操作
  1677.             if (abs(i) < (radius / 3) && abs(j) < (radius / 3))
  1678.             {
  1679.                 //这里rbin,cbin范围是(-1,d)
  1680.                 if (rbin > -1 && rbin < d && cbin>-1 && cbin < d &&
  1681.                     r>0 && r < rows - 1 && c>0 && c < cols - 1)
  1682.                 {
  1683.                     float dx = gauss_image.at<float>(r, c + 1) - gauss_image.at<float>(r, c - 1);
  1684.                     float dy = gauss_image.at<float>(r + 1, c) - gauss_image.at<float>(r - 1, c);
  1686.                     X[k1] = dx;                                //邻域内所有像素点的水平差分
  1687.                     Y[k1] = dy;                                //邻域内所有像素点的竖直差分
  1689.                     RBin[k1] = rbin;                        //邻域内所有采样点落在子区域的 y 坐标
  1690.                     CBin[k1] = cbin;                        //邻域内所有采样点落在子区域的 x 坐标
  1692.                     //高斯权值的指数部分
  1693.                     W[k1] = (c_rot * c_rot + r_rot * r_rot) * exp_scale;
  1695.                     ++k1;
  1696.                 }
  1697.             }
  1698.             //对离中心点远的部分进行操作
  1699.             else if (abs(i) < (2 * radius / 3) && abs(i) > (radius / 3) && abs(j) < (2 * radius / 3) && abs(j) > (radius / 3))
  1700.             {
  1701.                 //这里rbin,cbin范围是(-1,d)
  1702.                 if (rbin > -1 && rbin < d && cbin>-1 && cbin < d &&
  1703.                     r>0 && r < rows - 1 && c>0 && c < cols - 1)
  1704.                 {
  1705.                     float dx = gauss_image.at<float>(r, c + 1) - gauss_image.at<float>(r, c - 1);
  1706.                     float dy = gauss_image.at<float>(r + 1, c) - gauss_image.at<float>(r - 1, c);
  1708.                     X2[k2] = dx;                                //邻域内所有像素点的水平差分
  1709.                     Y2[k2] = dy;                                //邻域内所有像素点的竖直差分
  1711.                     RBin2[k2] = rbin;                            //邻域内所有采样点落在子区域的 y 坐标
  1712.                     CBin2[k2] = cbin;                            //邻域内所有采样点落在子区域的 x 坐标
  1714.                     //高斯权值的指数部分
  1715.                     W2[k2] = (c_rot * c_rot + r_rot * r_rot) * exp_scale;
  1717.                     ++k2;
  1718.                 }
  1719.             }
  1720.             else
  1721.             {
  1722.                 //这里rbin,cbin范围是(-1,d)
  1723.                 if (rbin > -1 && rbin < d && cbin>-1 && cbin < d &&
  1724.                     r>0 && r < rows - 1 && c>0 && c < cols - 1)
  1725.                 {
  1726.                     float dx = gauss_image.at<float>(r, c + 1) - gauss_image.at<float>(r, c - 1);
  1727.                     float dy = gauss_image.at<float>(r + 1, c) - gauss_image.at<float>(r - 1, c);
  1729.                     X3[k3] = dx;                                //邻域内所有像素点的水平差分
  1730.                     Y3[k3] = dy;                                //邻域内所有像素点的竖直差分
  1732.                     RBin3[k3] = rbin;                            //邻域内所有采样点落在子区域的 y 坐标
  1733.                     CBin3[k3] = cbin;                            //邻域内所有采样点落在子区域的 x 坐标
  1735.                     //高斯权值的指数部分
  1736.                     W3[k3] = (c_rot * c_rot + r_rot * r_rot) * exp_scale;
  1738.                     ++k3;
  1739.                 }
  1740.             }
  1741.         }
  1742.     }
  1744.     //两个区域内数组的合并拼接
  1745.     for (int k = 0; k < k2; k++)
  1746.     {
  1747.         X[k1 + k] = X2[k];
  1748.         Y[k1 + k] = Y2[k];
  1750.         RBin[k1 + k] = RBin2[k];
  1751.         CBin[k1 + k] = CBin2[k];
  1753.         W[k1 + k] = W2[k];
  1755.     }
  1757.     for (int k = 0; k < k3; k++)
  1758.     {
  1759.         X[k1 + k2 + k] = X3[k];
  1760.         Y[k1 + k2 + k] = Y3[k];
  1762.         RBin[k1 + k2 + k] = RBin3[k];
  1763.         CBin[k1 + k2 + k] = CBin3[k];
  1765.         W[k1 + k2 + k] = W3[k];
  1767.     }
  1769.     //计算采样点落在子区域的像素梯度幅度，梯度角度，和高斯权值
  1770.     len = k1 + k2 + k3;
  1772.     cv::hal::exp(W, W, len);                        //邻域内所有采样点落在子区域的像素的高斯权值
  1773.     cv::hal::fastAtan2(Y, X, Angle, len, true);        //邻域内所有采样点落在子区域的像素的梯度方向，角度范围是0-360度
  1774.     cv::hal::magnitude(X, Y, Mag, len);                //邻域内所有采样点落在子区域的像素的梯度幅度
  1776.     //计算每个特征点的特征描述子
  1777.     for (int k = 0; k < len; ++k)
  1778.     {
  1779.         float rbin = RBin[k], cbin = CBin[k];        //子区域内像素点坐标，rbin,cbin范围是(-1,d)
  1781.         //离特征点进的邻域
  1782.         if (k < k1)
  1783.         {
  1784.             //子区域内像素点处理后的方向
  1785.             float obin = (Angle[k] - main_angle) * bin_per_rad_1;
  1787.             float mag = Mag[k] * W[k];                    //子区域内像素点乘以权值后的梯度幅值
  1789.             int r0 = cvFloor(rbin);                        //ro取值集合是{-1,0,1,2，3}，向下取整
  1790.             int c0 = cvFloor(cbin);                        //c0取值集合是{-1，0，1，2，3}
  1791.             int o0 = cvFloor(obin);
  1793.             rbin = rbin - r0;                            //子区域内像素点坐标的小数部分，用于线性插值
  1794.             cbin = cbin - c0;
  1795.             obin = obin - o0;
  1797.             //限制范围为梯度直方图横坐标[0,n)
  1798.             if (o0 < 0)
  1799.                 o0 = o0 + n1;
  1800.             if (o0 >= n1)
  1801.                 o0 = o0 - n1;
  1803.             //三线性插值用于计算落在两个子区域之间的像素对两个子区域的作用，并把其分配到对应子区域的8个方向上
  1804.             //使用三线性插值(即三维)方法，计算直方图
  1805.             float v_r1 = mag * rbin;                    //第二行分配的值
  1806.             float v_r0 = mag - v_r1;                    //第一行分配的值
  1808.             float v_rc11 = v_r1 * cbin;                    //第二行第二列分配的值
  1809.             float v_rc10 = v_r1 - v_rc11;                //第二行第一列分配的值
  1811.             float v_rc01 = v_r0 * cbin;                    //第一行第二列分配的值
  1812.             float v_rc00 = v_r0 - v_rc01;
  1814.             float v_rco111 = v_rc11 * obin;                //第二行第二列第二个方向上分配的值，每个采样点去邻近两个方向
  1815.             float v_rco110 = v_rc11 - v_rco111;            //第二行第二列第一个方向上分配的值
  1817.             float v_rco101 = v_rc10 * obin;
  1818.             float v_rco100 = v_rc10 - v_rco101;
  1820.             float v_rco011 = v_rc01 * obin;
  1821.             float v_rco010 = v_rc01 - v_rco011;
  1823.             float v_rco001 = v_rc00 * obin;
  1824.             float v_rco000 = v_rc00 - v_rco001;
  1826.             //该像素所在网格的索引
  1827.             int idx = ((r0 + 1) * (d + 2) + c0 + 1) * (n1 + 2) + o0;
  1829.             hist[idx] += v_rco000;
  1830.             hist[idx + 1] += v_rco001;
  1831.             hist[idx + n1 + 2] += v_rco010;
  1832.             hist[idx + n1 + 3] += v_rco011;
  1833.             hist[idx + (d + 2) * (n1 + 2)] += v_rco100;
  1834.             hist[idx + (d + 2) * (n1 + 2) + 1] += v_rco101;
  1835.             hist[idx + (d + 3) * (n1 + 2)] += v_rco110;
  1836.             hist[idx + (d + 3) * (n1 + 2) + 1] += v_rco111;
  1837.         }
  1839.         //离特征点远的邻域
  1840.         else if (k >= k1 && k < k2)
  1841.         {
  1842.             //子区域内像素点处理后的方向
  1843.             float obin = (Angle[k] - main_angle) * bin_per_rad_2;
  1845.             float mag = Mag[k] * W[k];                    //子区域内像素点乘以权值后的梯度幅值
  1847.             int r0 = cvFloor(rbin);                        //ro取值集合是{-1,0,1,2，3}，向下取整
  1848.             int c0 = cvFloor(cbin);                        //c0取值集合是{-1，0，1，2，3}
  1849.             int o0 = cvFloor(obin);
  1851.             rbin = rbin - r0;                            //子区域内像素点坐标的小数部分，用于线性插值
  1852.             cbin = cbin - c0;
  1853.             obin = obin - o0;
  1855.             //限制范围为梯度直方图横坐标[0,n)
  1856.             if (o0 < 0)
  1857.                 o0 = o0 + n2;
  1858.             if (o0 >= n1)
  1859.                 o0 = o0 - n2;
  1861.             //三线性插值用于计算落在两个子区域之间的像素对两个子区域的作用，并把其分配到对应子区域的8个方向上
  1862.             //使用三线性插值(即三维)方法，计算直方图
  1863.             float v_r1 = mag * rbin;                    //第二行分配的值
  1864.             float v_r0 = mag - v_r1;                    //第一行分配的值
  1866.             float v_rc11 = v_r1 * cbin;                    //第二行第二列分配的值
  1867.             float v_rc10 = v_r1 - v_rc11;                //第二行第一列分配的值
  1869.             float v_rc01 = v_r0 * cbin;                    //第一行第二列分配的值
  1870.             float v_rc00 = v_r0 - v_rc01;
  1872.             float v_rco111 = v_rc11 * obin;                //第二行第二列第二个方向上分配的值，每个采样点去邻近两个方向
  1873.             float v_rco110 = v_rc11 - v_rco111;            //第二行第二列第一个方向上分配的值
  1875.             float v_rco101 = v_rc10 * obin;
  1876.             float v_rco100 = v_rc10 - v_rco101;
  1878.             float v_rco011 = v_rc01 * obin;
  1879.             float v_rco010 = v_rc01 - v_rco011;
  1881.             float v_rco001 = v_rc00 * obin;
  1882.             float v_rco000 = v_rc00 - v_rco001;
  1884.             //该像素所在网格的索引
  1885.             int idx = ((r0 + 1) * (d + 2) + c0 + 1) * (n2 + 2) + o0;
  1887.             hist[idx] += v_rco000;
  1888.             hist[idx + 1] += v_rco001;
  1889.             hist[idx + n2 + 2] += v_rco010;
  1890.             hist[idx + n2 + 3] += v_rco011;
  1891.             hist[idx + (d + 2) * (n2 + 2)] += v_rco100;
  1892.             hist[idx + (d + 2) * (n2 + 2) + 1] += v_rco101;
  1893.             hist[idx + (d + 3) * (n2 + 2)] += v_rco110;
  1894.             hist[idx + (d + 3) * (n2 + 2) + 1] += v_rco111;
  1895.         }
  1896.         else
  1897.         {
  1898.             //子区域内像素点处理后的方向
  1899.             float obin = (Angle[k] - main_angle) * bin_per_rad_3;
  1901.             float mag = Mag[k] * W[k];                    //子区域内像素点乘以权值后的梯度幅值
  1903.             int r0 = cvFloor(rbin);                        //ro取值集合是{-1,0,1,2，3}，向下取整
  1904.             int c0 = cvFloor(cbin);                        //c0取值集合是{-1，0，1，2，3}
  1905.             int o0 = cvFloor(obin);
  1907.             rbin = rbin - r0;                            //子区域内像素点坐标的小数部分，用于线性插值
  1908.             cbin = cbin - c0;
  1909.             obin = obin - o0;
  1911.             //限制范围为梯度直方图横坐标[0,n)
  1912.             if (o0 < 0)
  1913.                 o0 = o0 + n3;
  1914.             if (o0 >= n1)
  1915.                 o0 = o0 - n3;
  1917.             //三线性插值用于计算落在两个子区域之间的像素对两个子区域的作用，并把其分配到对应子区域的8个方向上
  1918.             //使用三线性插值(即三维)方法，计算直方图
  1919.             float v_r1 = mag * rbin;                    //第二行分配的值
  1920.             float v_r0 = mag - v_r1;                    //第一行分配的值
  1922.             float v_rc11 = v_r1 * cbin;                    //第二行第二列分配的值
  1923.             float v_rc10 = v_r1 - v_rc11;                //第二行第一列分配的值
  1925.             float v_rc01 = v_r0 * cbin;                    //第一行第二列分配的值
  1926.             float v_rc00 = v_r0 - v_rc01;
  1928.             float v_rco111 = v_rc11 * obin;                //第二行第二列第二个方向上分配的值，每个采样点去邻近两个方向
  1929.             float v_rco110 = v_rc11 - v_rco111;            //第二行第二列第一个方向上分配的值
  1931.             float v_rco101 = v_rc10 * obin;
  1932.             float v_rco100 = v_rc10 - v_rco101;
  1934.             float v_rco011 = v_rc01 * obin;
  1935.             float v_rco010 = v_rc01 - v_rco011;
  1937.             float v_rco001 = v_rc00 * obin;
  1938.             float v_rco000 = v_rc00 - v_rco001;
  1940.             //该像素所在网格的索引
  1941.             int idx = ((r0 + 1) * (d + 2) + c0 + 1) * (n3 + 2) + o0;
  1943.             hist[idx] += v_rco000;
  1944.             hist[idx + 1] += v_rco001;
  1945.             hist[idx + n3 + 2] += v_rco010;
  1946.             hist[idx + n3 + 3] += v_rco011;
  1947.             hist[idx + (d + 2) * (n3 + 2)] += v_rco100;
  1948.             hist[idx + (d + 2) * (n3 + 2) + 1] += v_rco101;
  1949.             hist[idx + (d + 3) * (n3 + 2)] += v_rco110;
  1950.             hist[idx + (d + 3) * (n3 + 2) + 1] += v_rco111;
  1951.         }
  1952.     }
  1954.     //由于圆周循环的特性，对计算以后幅角小于 0 度或大于 360 度的值重新进行调整，使
  1955.     //其在 0～360 度之间
  1956.     for (int i = 0; i < d; ++i)
  1957.     {
  1958.         for (int j = 0; j < d; ++j)
  1959.         {
  1960.             int idx = ((i + 1) * (d + 2) + (j + 1)) * (n + 2);
  1961.             hist[idx] += hist[idx + n];
  1962.             //hist[idx + 1] += hist[idx + n + 1];//opencv源码中这句话是多余的,hist[idx + n + 1]永远是0.0
  1963.             for (int k = 0; k < n; ++k)
  1964.                 descriptor[(i * d + j) * n + k] = hist[idx + k];
  1965.         }
  1966.     }
  1968.     //对描述子进行归一化
  1969.     int lenght = d * d * n;
  1970.     float norm = 0;
  1972.     //计算特征描述向量的模值的平方
  1973.     for (int i = 0; i < lenght; ++i)
  1974.     {
  1975.         norm = norm + descriptor[i] * descriptor[i];
  1976.     }
  1977.     norm = sqrt(norm);                            //特征描述向量的模值
  1979.     //此次归一化能去除光照的影响
  1980.     for (int i = 0; i < lenght; ++i)
  1981.     {
  1982.         descriptor[i] = descriptor[i] / norm;
  1983.     }
  1985.     //阈值截断,去除特征描述向量中大于 0.2 的值，能消除非线性光照的影响(相机饱和度对某些放的梯度影响较大，对方向的影响较小)
  1986.     for (int i = 0; i < lenght; ++i)
  1987.     {
  1988.         descriptor[i] = min(descriptor[i], DESCR_MAG_THR);
  1989.     }
  1991.     //再次归一化，能够提高特征的独特性
  1992.     norm = 0;
  1994.     for (int i = 0; i < lenght; ++i)
  1995.     {
  1996.         norm = norm + descriptor[i] * descriptor[i];
  1997.     }
  1998.     norm = sqrt(norm);
  1999.     for (int i = 0; i < lenght; ++i)
  2000.     {
  2001.         descriptor[i] = descriptor[i] / norm;
  2002.     }
  2003. }
  2005. /********************************该函数计算所有特征点特征描述子***************************/
  2006. /*gauss_pyr表示高斯金字塔
  2007.   keypoints表示特征点、
  2008.   descriptors表示生成的特征点的描述子*/
  2009. void mySift::calc_sift_descriptors(const vector<vector<Mat>>& gauss_pyr, vector<KeyPoint>& keypoints,
  2010.     Mat& descriptors, const vector<Mat>& amplit, const vector<Mat>& orient)
  2011. {
  2012.     int d = DESCR_WIDTH;                                        //d=4,特征点邻域网格个数是d x d
  2013.     int n = DESCR_HIST_BINS;                                    //n=8,每个网格特征点梯度角度等分为8个方向
  2015.     descriptors.create(keypoints.size(), d * d * n, CV_32FC1);    //分配空间
  2017.     for (size_t i = 0; i < keypoints.size(); ++i)                //对于每一个特征点
  2018.     {
  2019.         int octaves, layer;
  2021.         //得到特征点所在的组号，层号
  2022.         octaves = keypoints[i].octave & 255;
  2023.         layer = (keypoints[i].octave >> 8) & 255;
  2025.         //得到特征点相对于本组的坐标，不是最底层
  2026.         Point2f pt(keypoints[i].pt.x / (1 << octaves), keypoints[i].pt.y / (1 << octaves));
  2028.         float scale = keypoints[i].size / (1 << octaves);        //得到特征点相对于本组的尺度
  2029.         float main_angle = keypoints[i].angle;                    //特征点主方向
  2031.         //计算该点的描述子
  2032.         calc_sift_descriptor(gauss_pyr[octaves][layer], main_angle, pt, scale, d, n, descriptors.ptr<float>((int)i));
  2033.         //improve_calc_sift_descriptor(gauss_pyr[octaves][layer], main_angle, pt, scale, d, n, descriptors.ptr<float>((int)i));
  2034.         //calc_gloh_descriptor(amplit[octaves], orient[octaves], pt, scale, main_angle, d, n, descriptors.ptr<float>((int)i));
  2036.         if (double_size)//如果图像尺寸扩大一倍
  2037.         {
  2038.             keypoints[i].pt.x = keypoints[i].pt.x / 2.f;
  2039.             keypoints[i].pt.y = keypoints[i].pt.y / 2.f;
  2040.         }
  2041.     }
  2043. }
  2045. /*************该函数构建SAR_SIFT尺度空间*****************/
  2046. /*image表示输入的原始图像
  2047.  sar_harris_fun表示尺度空间的Sar_harris函数
  2048.  amplit表示尺度空间像素的梯度幅度
  2049.  orient表示尺度空间像素的梯度方向
  2050.  */
  2051. void mySift::build_sar_sift_space(const Mat& image, vector<Mat>& sar_harris_fun, vector<Mat>& amplit, vector<Mat>& orient)
  2052. {
  2053.     //转换输入图像格式
  2054.     Mat gray_image;
  2055.     if (image.channels() != 1)
  2056.         cvtColor(image, gray_image, CV_RGB2GRAY);
  2057.     else
  2058.         gray_image = image;
  2060.     //把图像转换为0-1之间的浮点数据
  2061.     Mat float_image;
  2063.     double ratio = pow(2, 1.0 / 3.0);                            //相邻两层的尺度比,默认是2^(1/3)
  2065.     //在这里转换为0-1之间的浮点数据和转换为0-255之间的浮点数据，效果是一样的
  2066.     //gray_image.convertTo(float_image, CV_32FC1, 1.f / 255.f, 0.f);//转换为0-1之间
  2068.     gray_image.convertTo(float_image, CV_32FC1, 1, 0.f);        //转换为0-255之间的浮点数
  2070.     //分配内存
  2071.     sar_harris_fun.resize(Mmax);
  2072.     amplit.resize(Mmax);
  2073.     orient.resize(Mmax);
  2075.     for (int i = 0; i < Mmax; ++i)
  2076.     {
  2077.         float scale = (float)sigma * (float)pow(ratio, i);        //获得当前层的尺度
  2078.         int radius = cvRound(2 * scale);
  2079.         Mat kernel;
  2081.         roewa_kernel(radius, scale, kernel);
  2083.         //四个滤波模板生成
  2084.         Mat W34 = Mat::zeros(2 * radius + 1, 2 * radius + 1, CV_32FC1);
  2085.         Mat W12 = Mat::zeros(2 * radius + 1, 2 * radius + 1, CV_32FC1);
  2086.         Mat W14 = Mat::zeros(2 * radius + 1, 2 * radius + 1, CV_32FC1);
  2087.         Mat W23 = Mat::zeros(2 * radius + 1, 2 * radius + 1, CV_32FC1);
  2089.         kernel(Range(radius + 1, 2 * radius + 1), Range::all()).copyTo(W34(Range(radius + 1, 2 * radius + 1), Range::all()));
  2090.         kernel(Range(0, radius), Range::all()).copyTo(W12(Range(0, radius), Range::all()));
  2091.         kernel(Range::all(), Range(radius + 1, 2 * radius + 1)).copyTo(W14(Range::all(), Range(radius + 1, 2 * radius + 1)));
  2092.         kernel(Range::all(), Range(0, radius)).copyTo(W23(Range::all(), Range(0, radius)));
  2094.         //滤波
  2095.         Mat M34, M12, M14, M23;
  2096.         double eps = 0.00001;
  2097.         filter2D(float_image, M34, CV_32FC1, W34, Point(-1, -1), eps);
  2098.         filter2D(float_image, M12, CV_32FC1, W12, Point(-1, -1), eps);
  2099.         filter2D(float_image, M14, CV_32FC1, W14, Point(-1, -1), eps);
  2100.         filter2D(float_image, M23, CV_32FC1, W23, Point(-1, -1), eps);
  2102.         //计算水平梯度和竖直梯度
  2103.         Mat Gx, Gy;
  2104.         log((M14) / (M23), Gx);
  2105.         log((M34) / (M12), Gy);
  2107.         //计算梯度幅度和梯度方向
  2108.         magnitude(Gx, Gy, amplit[i]);
  2109.         phase(Gx, Gy, orient[i], true);
  2111.         //构建sar-Harris矩阵
  2112.         //Mat Csh_11 = log(scale)*log(scale)*Gx.mul(Gx);
  2113.         //Mat Csh_12 = log(scale)*log(scale)*Gx.mul(Gy);
  2114.         //Mat Csh_22 = log(scale)*log(scale)*Gy.mul(Gy);
  2116.         Mat Csh_11 = scale * scale * Gx.mul(Gx);
  2117.         Mat Csh_12 = scale * scale * Gx.mul(Gy);
  2118.         Mat Csh_22 = scale * scale * Gy.mul(Gy);//此时阈值为0.8
  2120.         //Mat Csh_11 = Gx.mul(Gx);
  2121.         //Mat Csh_12 = Gx.mul(Gy);
  2122.         //Mat Csh_22 = Gy.mul(Gy);//此时阈值为0.8/100
  2124.         //高斯权重
  2125.         float gauss_sigma = sqrt(2.f) * scale;
  2126.         int size = cvRound(3 * gauss_sigma);
  2128.         Size kern_size(2 * size + 1, 2 * size + 1);
  2129.         GaussianBlur(Csh_11, Csh_11, kern_size, gauss_sigma, gauss_sigma);
  2130.         GaussianBlur(Csh_12, Csh_12, kern_size, gauss_sigma, gauss_sigma);
  2131.         GaussianBlur(Csh_22, Csh_22, kern_size, gauss_sigma, gauss_sigma);
  2133.         /*Mat gauss_kernel;//自定义圆形高斯核
  2134.         gauss_circle(size, gauss_sigma, gauss_kernel);
  2135.         filter2D(Csh_11, Csh_11, CV_32FC1, gauss_kernel);
  2136.         filter2D(Csh_12, Csh_12, CV_32FC1, gauss_kernel);
  2137.         filter2D(Csh_22, Csh_22, CV_32FC1, gauss_kernel);*/
  2139.         Mat Csh_21 = Csh_12;
  2141.         //构建sar_harris函数
  2142.         Mat temp_add = Csh_11 + Csh_22;
  2144.         double d = 0.04;                    //sar_haiirs函数表达式中的任意参数，默认是0.04
  2146.         sar_harris_fun[i] = Csh_11.mul(Csh_22) - Csh_21.mul(Csh_12) - (float)d * temp_add.mul(temp_add);
  2147.     }
  2148. }
  2150. /***************该函数计算所有特征点的特征向量*************/
  2151. /*amplit表示尺度空间像素幅度
  2152.  orient表示尺度空间像素梯度角度
  2153.  keys表示检测到的特征点
  2154.  descriptors表示特征点描述子向量，【M x N】,M表示描述子个数，N表示描述子维度
  2155.  */
  2156. void mySift::calc_gloh_descriptors(const vector<Mat>& amplit, const vector<Mat>& orient, const vector<KeyPoint>& keys, Mat& descriptors)
  2157. {
  2158.     int d = SAR_SIFT_GLOH_ANG_GRID;                            //d=4或者d=8
  2159.     int n = SAR_SIFT_DES_ANG_BINS;                            //n=8默认
  2161.     int num_keys = (int)keys.size();
  2162.     int grids = 2 * d + 1;
  2164.     //descriptors.create(num_keys, grids * n, CV_32FC1);
  2165.     descriptors.create(num_keys, grids * n, CV_32FC1);
  2167.     for (int i = 0; i < num_keys; ++i)
  2168.     {
  2169.         int octaves = keys[i].octave & 255;                    //特征点所在层
  2171.         float* ptr_des = descriptors.ptr<float>(i);
  2172.         float scale = keys[i].size / (1 << octaves);        //得到特征点相对于本组的尺度;                            //特征点所在层的尺度
  2173.         float main_ori = keys[i].angle;                        //特征点主方向
  2175.         //得到特征点相对于本组的坐标，不是最底层
  2176.         Point2f point(keys[i].pt.x / (1 << octaves), keys[i].pt.y / (1 << octaves));
  2178.         cout << "layer=" << octaves << endl;
  2179.         cout << "scale=" << scale << endl;
  2181.         //计算该特征点的特征描述子
  2182.         calc_gloh_descriptor(amplit[octaves], orient[octaves], point, scale, main_ori, d, n, ptr_des);
  2183.     }
  2184. }
  2186. //特征点检测和特征点描述把整个 SIFT 算子都涵盖在内了//
  2188. /******************************特征点检测*********************************/
  2189. /*image表示输入的图像
  2190.   gauss_pyr表示生成的高斯金字塔
  2191.   dog_pyr表示生成的高斯差分DOG金字塔
  2192.   keypoints表示检测到的特征点
  2193.   vector<float>& cell_contrast 用于存放一个单元格中所有特征点的对比度
  2194.   vector<float>& cell_contrasts用于存放一个尺度层中所有单元格中特征点的对比度
  2195.   vector<vector<vector<float>>>& all_cell_contrasts用于存放所有尺度层中所有单元格的对比度
  2196.   vector<vector<float>>& average_contrast用于存放所有尺度层中多有单元格的平均对比度*/
  2197. void mySift::detect(const Mat& image, vector<vector<Mat>>& gauss_pyr, vector<vector<Mat>>& dog_pyr, vector<KeyPoint>& keypoints,
  2198.     vector<vector<vector<float>>>& all_cell_contrasts,
  2199.     vector<vector<float>>& average_contrast, vector<vector<int>>& n_cells, vector<int>& num_cell, vector<vector<int>>& available_n,
  2200.     vector<int>& available_num, vector<KeyPoint>& final_keypoints,
  2201.     vector<KeyPoint>& Final_keypoints, vector<KeyPoint>& Final_Keypoints)
  2202. {
  2203.     if (image.empty() || image.depth() != CV_8U)
  2204.         CV_Error(CV_StsBadArg, "输入图像为空，或者图像深度不是CV_8U");
  2206.     //计算高斯金字塔组数
  2207.     int nOctaves;
  2208.     nOctaves = num_octaves(image);
  2210.     //生成高斯金字塔第一层图像
  2211.     Mat init_gauss;
  2212.     create_initial_image(image, init_gauss);
  2214.     //生成高斯尺度空间图像
  2215.     build_gaussian_pyramid(init_gauss, gauss_pyr, nOctaves);
  2217.     //生成高斯差分金字塔(DOG金字塔，or LOG金字塔)
  2218.     build_dog_pyramid(dog_pyr, gauss_pyr);
  2220.     //在DOG金字塔上检测特征点
  2221.     find_scale_space_extrema1(dog_pyr, gauss_pyr, keypoints);            //原始 SIFT 算子
  2222.                                                                         // UR_SIFT,仅仅是检测特征点，并未对不理想的点进行筛选
  2223.     /*UR_SIFT_find_scale_space_extrema(dog_pyr, gauss_pyr, keypoints, all_cell_contrasts,
  2224.     average_contrast, n_cells, num_cell, available_n, available_num, final_keypoints, Final_keypoints, Final_Keypoints, N);*/
  2226.     //如果指定了特征点个数,并且设定的设置小于默认检测的特征点个数
  2227.     if (nfeatures != 0 && nfeatures < (int)keypoints.size())
  2228.     {
  2229.         //特征点响应值(即对比度，对比度越小越不稳定)从大到小排序
  2230.         sort(keypoints.begin(), keypoints.end(), [](const KeyPoint& a, const KeyPoint& b) {return a.response > b.response; });
  2232.         //删除点多余的特征点
  2233.         keypoints.erase(keypoints.begin() + nfeatures, keypoints.end());
  2234.     }
  2235. }
  2237. /**********************特征点描述*******************/
  2238. /*gauss_pyr表示高斯金字塔
  2239.  keypoints表示特征点集合
  2240.  descriptors表示特征点的描述子*/
  2241. void mySift::comput_des(const vector<vector<Mat>>& gauss_pyr, vector<KeyPoint>& final_keypoints, const vector<Mat>& amplit,
  2242.     const vector<Mat>& orient, Mat& descriptors)
  2243. {
  2244.     calc_sift_descriptors(gauss_pyr, final_keypoints, descriptors, amplit, orient);
  2245. }

3、 myMatch.h

  1. #pragma once
  3. #include<opencv2\core\core.hpp>
  4. #include<opencv2\features2d\features2d.hpp>
  6. #include<vector>
  7. #include<string>
  8. #include<iostream>
  10. const double dis_ratio1 = 0.75;            //最近邻和次近邻距离比阈值，就目前测试来看 dis_ratio = 0.75 时正确匹配的数量相对较多
  11. const double dis_ratio2 = 0.8;
  12. const double dis_ratio3 = 0.9;
  14. const float ransac_error = 1.5;            //ransac算法误差阈值
  16. const double FSC_ratio_low = 0.8;
  18. const double FSC_ratio_up = 1;
  20. const int pointsCount = 9;                // k均值聚类数据点个数
  21. const int clusterCount = 3;                // k均值聚类质心的个数
  23. enum DIS_CRIT { Euclidean = 0, COS };    //距离度量准则
  25. using namespace std;
  26. using namespace cv;
  28. class myMatch
  29. {
  30. public:
  32.     myMatch();
  33.     ~myMatch();
  35.     //该函数根据正确的匹配点对，计算出图像之间的变换关系
  36.     Mat LMS(const Mat& match1_xy, const Mat& match2_xy, string model, float& rmse);
  37.     
  38.     //改进版LMS超定方程
  39.     Mat improve_LMS(const Mat& match1_xy, const Mat& match2_xy, string model, float& rmse);
  41.     //该函数删除错误的匹配点对
  42.     Mat ransac(const vector<Point2f>& points_1, const vector<Point2f>& points_2, string model, float threshold, vector<bool>& inliers, float& rmse);
  44.     //绘制棋盘图像
  45.     void mosaic_map(const Mat& image_1, const Mat& image_2, Mat& chessboard_1, Mat& chessboard_2, Mat& mosaic_image, int width);
  47.     //该函数把两幅配准后的图像进行融合镶嵌
  48.     void image_fusion(const Mat& image_1, const Mat& image_2, const Mat T, Mat& fusion_image, Mat& matched_image);
  50.     //该函数进行描述子的最近邻和次近邻匹配
  51.     void match_des(const Mat& des_1, const Mat& des_2, vector<vector<DMatch>>& dmatchs, DIS_CRIT dis_crite);
  53.     //建立尺度直方图、ROM 直方图
  54.     void scale_ROM_Histogram(const vector<DMatch>& matches, float* scale_hist, float* ROM_hist, int n);
  56.     //该函数删除错误匹配点对，并进行配准
  57.     Mat match(const Mat& image_1, const Mat& image_2, const vector<vector<DMatch>>& dmatchs, vector<KeyPoint> keys_1,
  58.         vector<KeyPoint> keys_2, string model, vector<DMatch>& right_matchs, Mat& matched_line, vector<DMatch>& init_matchs);
  59. };

4、 myMatch.cpp

  1. #include"myMatch.h"
  3. #include<opencv2\imgproc\types_c.h>
  4. #include<opencv2\imgproc\imgproc.hpp>
  5. #include<opencv2\highgui\highgui.hpp>
  6. #include<opencv2\features2d\features2d.hpp>    //特征提取
  8. #include<algorithm>
  9. #include<vector>
  10. #include<cmath>
  11. #include<opencv.hpp>
  13. #include <numeric>                            //用于容器元素求和
  15. using namespace std;
  16. using namespace cv;
  17. //using namespace gpu;
  19. RNG rng(100);
  21. myMatch::myMatch()
  22. {
  23. }
  25. /********该函数根据正确的匹配点对，计算出图像之间的变换关系********/
  26. /*注意：输入几个点都能计算对应的 x 矩阵，(2N,8)*(8,1)=(2N,1)
  27.  match1_xy表示参考图像特征点坐标集合,[M x 2]矩阵，M表示特征的个数
  28.  match2_xy表示待配准图像特征点集合，[M x 2]矩阵，M表示特征点集合
  29.  model表示变换类型，“相似变换”,"仿射变换","透视变换"
  30.  rmse表示均方根误差
  31.  返回值为计算得到的3 x 3矩阵参数
  32.  */
  33. Mat myMatch::LMS(const Mat& match1_xy, const Mat& match2_xy, string model, float& rmse)
  34. {
  36.     if (match1_xy.rows != match2_xy.rows)
  37.         CV_Error(CV_StsBadArg, "LMS模块输入特征点对个数不一致！");
  39.     if (!(model == string("affine") || model == string("similarity") ||
  40.         model == string("perspective") || model == string("projective")))
  41.         CV_Error(CV_StsBadArg, "LMS模块图像变换类型输入错误！");
  43.     const int N = match1_xy.rows;                            //特征点个数
  45.     Mat match2_xy_trans, match1_xy_trans;                    //特征点坐标转置
  47.     transpose(match1_xy, match1_xy_trans);                    //矩阵转置(2,M)
  48.     transpose(match2_xy, match2_xy_trans);
  50.     Mat change = Mat::zeros(3, 3, CV_32FC1);                //变换矩阵
  52.     //A*X=B,接下来部分仿射变换和透视变换一样,如果特征点个数是M，则A=[2*M,6]矩阵
  53.     //A=[x1,y1,0,0,1,0;0,0,x1,y1,0,1;.....xn,yn,0,0,1,0;0,0,xn,yn,0,1]，应该是改版过的
  54.     Mat A = Mat::zeros(2 * N, 6, CV_32FC1);
  56.     for (int i = 0; i < N; ++i)
  57.     {
  58.         A.at<float>(2 * i, 0) = match2_xy.at<float>(i, 0);//x
  59.         A.at<float>(2 * i, 1) = match2_xy.at<float>(i, 1);//y
  60.         A.at<float>(2 * i, 4) = 1.f;
  62.         A.at<float>(2 * i + 1, 2) = match2_xy.at<float>(i, 0);
  63.         A.at<float>(2 * i + 1, 3) = match2_xy.at<float>(i, 1);
  64.         A.at<float>(2 * i + 1, 5) = 1.f;
  65.     }
  67.     //如果特征点个数是M,那个B=[2*M,1]矩阵
  68.     //B=[u1,v1,u2,v2,.....,un,vn]
  69.     Mat B;
  71.     B.create(2 * N, 1, CV_32FC1);
  72.     for (int i = 0; i < N; ++i)
  73.     {
  74.         B.at<float>(2 * i, 0) = match1_xy.at<float>(i, 0);      //x
  75.         B.at<float>(2 * i + 1, 0) = match1_xy.at<float>(i, 1);//y
  76.     }
  78.     //如果是仿射变换
  79.     if (model == string("affine"))
  80.     {
  81.         Vec6f values;
  82.         solve(A, B, values, DECOMP_QR);
  83.         change = (Mat_<float>(3, 3) << values(0), values(1), values(4),
  84.             values(2), values(3), values(5),
  85.             +0.0f, +0.0f, 1.0f);
  87.         Mat temp_1 = change(Range(0, 2), Range(0, 2));        //尺度和旋转量
  88.         Mat temp_2 = change(Range(0, 2), Range(2, 3));        //平移量
  90.         Mat match2_xy_change = temp_1 * match2_xy_trans + repeat(temp_2, 1, N);
  91.         Mat diff = match2_xy_change - match1_xy_trans;        //求差
  92.         pow(diff, 2.f, diff);
  93.         rmse = (float)sqrt(sum(diff)(0) * 1.0) / N;            //sum输出是各个通道的和, / N 初始实在括号里面
  94.     }
  96.     //如果是透视变换
  97.     else if (model == string("perspective"))
  98.     {
  99.         /*透视变换模型
  100.         [u'*w,v'*w, w]'=[u,v,w]' = [a1, a2, a5;
  101.                                     a3, a4, a6;
  102.                                     a7, a8, 1] * [x, y, 1]'
  103.         [u',v']'=[x,y,0,0,1,0,-u'x, -u'y;
  104.                  0, 0, x, y, 0, 1, -v'x,-v'y] * [a1, a2, a3, a4, a5, a6, a7, a8]'
  105.         即，Y = A*X     */
  107.         //构造 A 矩阵的后两列
  108.         Mat A2;
  109.         A2.create(2 * N, 2, CV_32FC1);
  110.         for (int i = 0; i < N; ++i)
  111.         {
  112.             A2.at<float>(2 * i, 0) = match1_xy.at<float>(i, 0) * match2_xy.at<float>(i, 0) * (-1.f);
  113.             A2.at<float>(2 * i, 1) = match1_xy.at<float>(i, 0) * match2_xy.at<float>(i, 1) * (-1.f);
  115.             A2.at<float>(2 * i + 1, 0) = match1_xy.at<float>(i, 1) * match2_xy.at<float>(i, 0) * (-1.f);
  116.             A2.at<float>(2 * i + 1, 1) = match1_xy.at<float>(i, 1) * match2_xy.at<float>(i, 1) * (-1.f);
  117.         }
  119.         Mat A1;                                            //完成的 A 矩阵，(8,8)
  120.         A1.create(2 * N, 8, CV_32FC1);
  121.         A.copyTo(A1(Range::all(), Range(0, 6)));
  122.         A2.copyTo(A1(Range::all(), Range(6, 8)));
  124.         Mat values;                                        //values中存放的是待求的8个参数
  125.         solve(A1, B, values, DECOMP_QR);                //(2N,8)*(8,1)=(2N,1)好像本身就有点超定矩阵的意思
  126.         change.at<float>(0, 0) = values.at<float>(0);
  127.         change.at<float>(0, 1) = values.at<float>(1);
  128.         change.at<float>(0, 2) = values.at<float>(4);
  129.         change.at<float>(1, 0) = values.at<float>(2);
  130.         change.at<float>(1, 1) = values.at<float>(3);
  131.         change.at<float>(1, 2) = values.at<float>(5);
  132.         change.at<float>(2, 0) = values.at<float>(6);
  133.         change.at<float>(2, 1) = values.at<float>(7);
  134.         change.at<float>(2, 2) = 1.f;
  136.         Mat temp1 = Mat::ones(1, N, CV_32FC1);
  137.         Mat temp2;
  138.         temp2.create(3, N, CV_32FC1);
  140.         match2_xy_trans.copyTo(temp2(Range(0, 2), Range::all()));
  141.         temp1.copyTo(temp2(Range(2, 3), Range::all()));
  143.         Mat match2_xy_change = change * temp2;           //待配准图像中的特征点在参考图中的映射结果
  144.         Mat match2_xy_change_12 = match2_xy_change(Range(0, 2), Range::all());
  145.         //float* temp_ptr = match2_xy_change.ptr<float>(2);
  147.         float* temp_ptr = match2_xy_change.ptr<float>(2);
  149.         for (int i = 0; i < N; ++i)
  150.         {
  151.             float div_temp = temp_ptr[i];
  152.             match2_xy_change_12.at<float>(0, i) = match2_xy_change_12.at<float>(0, i) / div_temp;
  153.             match2_xy_change_12.at<float>(1, i) = match2_xy_change_12.at<float>(1, i) / div_temp;
  154.         }
  156.         Mat diff = match2_xy_change_12 - match1_xy_trans;
  157.         pow(diff, 2, diff);
  159.         rmse = (float)sqrt(sum(diff)(0) * 1.0) / N;      //sum输出是各个通道的和，rmse为输入点的均方误差
  160.     }
  162.     //如果是相似变换
  163.     else if (model == string("similarity"))
  164.     {
  165.         /*[x, y, 1, 0;
  166.           y, -x, 0, 1] * [a, b, c, d]'=[u,v]*/
  168.         Mat A3;
  169.         A3.create(2 * N, 4, CV_32FC1);
  170.         for (int i = 0; i < N; ++i)
  171.         {
  172.             A3.at<float>(2 * i, 0) = match2_xy.at<float>(i, 0);
  173.             A3.at<float>(2 * i, 1) = match2_xy.at<float>(i, 1);
  174.             A3.at<float>(2 * i, 2) = 1.f;
  175.             A3.at<float>(2 * i, 3) = 0.f;
  177.             A3.at<float>(2 * i + 1, 0) = match2_xy.at<float>(i, 1);
  178.             A3.at<float>(2 * i + 1, 1) = match2_xy.at<float>(i, 0) * (-1.f);
  179.             A3.at<float>(2 * i + 1, 2) = 0.f;
  180.             A3.at<float>(2 * i + 1, 3) = 1.f;
  181.         }
  183.         Vec4f values;
  184.         solve(A3, B, values, DECOMP_QR);
  185.         change = (Mat_<float>(3, 3) << values(0), values(1), values(2),
  186.             values(1) * (-1.0f), values(0), values(3),
  187.             +0.f, +0.f, 1.f);
  189.         Mat temp_1 = change(Range(0, 2), Range(0, 2));//尺度和旋转量
  190.         Mat temp_2 = change(Range(0, 2), Range(2, 3));//平移量
  192.         Mat match2_xy_change = temp_1 * match2_xy_trans + repeat(temp_2, 1, N);
  193.         Mat diff = match2_xy_change - match1_xy_trans;//求差
  194.         pow(diff, 2, diff);
  195.         rmse = (float)sqrt(sum(diff)(0) * 1.0) / N;//sum输出是各个通道的和
  196.     }
  198.     return change;
  199. }
  201. /********改进版LMS超定方程********/
  202. Mat myMatch::improve_LMS(const Mat& match1_xy, const Mat& match2_xy, string model, float& rmse)
  203. {
  205.     if (match1_xy.rows != match2_xy.rows)
  206.         CV_Error(CV_StsBadArg, "LMS模块输入特征点对个数不一致！");
  208.     if (!(model == string("affine") || model == string("similarity") ||
  209.         model == string("perspective")))
  210.         CV_Error(CV_StsBadArg, "LMS模块图像变换类型输入错误！");
  212.     const int N = match1_xy.rows;                            //特征点个数
  214.     Mat match2_xy_trans, match1_xy_trans;                    //特征点坐标转置
  216.     transpose(match1_xy, match1_xy_trans);                    //矩阵转置(2,M)
  217.     transpose(match2_xy, match2_xy_trans);
  219.     Mat change = Mat::zeros(3, 3, CV_32FC1);                //变换矩阵
  221.     //A*X=B,接下来部分仿射变换和透视变换一样,如果特征点个数是M，则A=[2*M,6]矩阵
  222.     //A=[x1,y1,0,0,1,0;0,0,x1,y1,0,1;.....xn,yn,0,0,1,0;0,0,xn,yn,0,1]，应该是改版过的
  223.     Mat A = Mat::zeros(2 * N, 6, CV_32FC1);
  225.     for (int i = 0; i < N; ++i)
  226.     {
  227.         A.at<float>(2 * i, 0) = match2_xy.at<float>(i, 0);//x
  228.         A.at<float>(2 * i, 1) = match2_xy.at<float>(i, 1);//y
  229.         A.at<float>(2 * i, 4) = 1.f;
  231.         //A.at<float>(2 * i, 0) = match2_xy.at<float>(i, 0);//x
  232.         //A.at<float>(2 * i, 1) = match2_xy.at<float>(i, 1);//y
  233.         //A.at<float>(2 * i, 2) = 1.f;
  235.         A.at<float>(2 * i + 1, 2) = match2_xy.at<float>(i, 0);
  236.         A.at<float>(2 * i + 1, 3) = match2_xy.at<float>(i, 1);
  237.         A.at<float>(2 * i + 1, 5) = 1.f;
  239.         /*A.at<float>(2 * i + 1, 3) = match2_xy.at<float>(i, 0);
  240.         A.at<float>(2 * i + 1, 4) = match2_xy.at<float>(i, 1);
  241.         A.at<float>(2 * i + 1, 5) = 1.f;*/
  242.     }
  244.     //如果特征点个数是M,那个B=[2*M,1]矩阵
  245.     //B=[u1,v1,u2,v2,.....,un,vn]
  246.     Mat B;
  248.     B.create(2 * N, 1, CV_32FC1);
  249.     for (int i = 0; i < N; ++i)
  250.     {
  251.         B.at<float>(2 * i, 0) = match1_xy.at<float>(i, 0);      //x
  252.         B.at<float>(2 * i + 1, 0) = match1_xy.at<float>(i, 1);//y
  253.     }
  255.     //如果是仿射变换
  256.     if (model == string("affine"))
  257.     {
  258.         Vec6f values;
  259.         solve(A, B, values, DECOMP_QR);
  260.         change = (Mat_<float>(3, 3) << values(0), values(1), values(4),
  261.             values(2), values(3), values(5),
  262.             +0.0f, +0.0f, 1.0f);
  264.         Mat temp_1 = change(Range(0, 2), Range(0, 2));//尺度和旋转量
  265.         Mat temp_2 = change(Range(0, 2), Range(2, 3));//平移量
  267.         Mat match2_xy_change = temp_1 * match2_xy_trans + repeat(temp_2, 1, N);
  268.         Mat diff = match2_xy_change - match1_xy_trans;//求差
  269.         pow(diff, 2.f, diff);
  270.         rmse = (float)sqrt(sum(diff)(0) * 1.0 / N);//sum输出是各个通道的和
  271.     }
  272.     //如果是透视变换
  273.     else if (model == string("perspective"))
  274.     {
  275.         /*透视变换模型
  276.         [u'*w,v'*w, w]'=[u,v,w]' = [a1, a2, a5;
  277.                                     a3, a4, a6;
  278.                                     a7, a8, 1] * [x, y, 1]'
  279.         [u',v']'=[x,y,0,0,1,0,-u'x, -u'y;
  280.                  0, 0, x, y, 0, 1, -v'x,-v'y] * [a1, a2, a3, a4, a5, a6, a7, a8]'
  281.         即，Y = A*X     */
  283.         //构造 A 矩阵的后两列
  284.         Mat A2;
  285.         A2.create(2 * N, 2, CV_32FC1);
  286.         for (int i = 0; i < N; ++i)
  287.         {
  288.             A2.at<float>(2 * i, 0) = match1_xy.at<float>(i, 0) * match2_xy.at<float>(i, 0) * (-1.f);
  289.             A2.at<float>(2 * i, 1) = match1_xy.at<float>(i, 0) * match2_xy.at<float>(i, 1) * (-1.f);
  291.             A2.at<float>(2 * i + 1, 0) = match1_xy.at<float>(i, 1) * match2_xy.at<float>(i, 0) * (-1.f);
  292.             A2.at<float>(2 * i + 1, 1) = match1_xy.at<float>(i, 1) * match2_xy.at<float>(i, 1) * (-1.f);
  293.         }
  295.         Mat A1;                                             //完整的 A 矩阵，(8,8)
  296.         A1.create(2 * N, 8, CV_32FC1);
  297.         A.copyTo(A1(Range::all(), Range(0, 6)));
  298.         A2.copyTo(A1(Range::all(), Range(6, 8)));
  300.         Mat AA1, balance;
  301.         Mat evects, evals;
  302.         transpose(A1, AA1);                                 //求矩阵 A1 的转置
  303.         balance = AA1 * A1;
  305.         double a[8][8];
  306.         for (int i = 0; i < 8; i++)
  307.         {
  308.             for (int j = 0; j < 8; j++)
  309.             {
  310.                 a[i][j] = balance.at<float>(i, j);
  311.             }
  312.         }
  313.         //构造输入矩阵
  314.         CvMat SrcMatrix = cvMat(8, 8, CV_32FC1, a);
  316.         double b[8][8] =
  317.         {
  318.            {0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0},
  319.            {0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0},
  320.            {0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0},
  321.            {0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0},
  322.            {0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0},
  323.            {0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0},
  324.            {0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0},
  325.            {0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0}
  326.         };
  327.         // 构造输出特征向量矩阵,特征向量按行存储,且与特征值相对应
  328.         CvMat ProVector = cvMat(8, 8, CV_32FC1, b);
  330.         // 构造输出特征值矩阵,特征值按降序配列
  331.         double c[8] = { 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0, 0.0 };
  332.         CvMat ProValue = cvMat(8, 1, CV_32FC1, c);
  334.         //求特征向量，最后一行特征向量即对应的 H 矩阵的参数
  335.         cvEigenVV(&SrcMatrix, &ProVector, &ProValue, 1.0e-6F);
  337.         输出特征向量矩阵
  338.         //for (int i = 0; i < 2; i++)
  339.         //{
  340.         //    for (int j = 0; j < 2; j++)
  341.         //        printf("%f\t", cvmGet(&ProVector, i, j));
  342.         //    printf("\n");
  343.         //}
  345.         //把计算得到的最小特征值对应的特征变量赋给 H 矩阵
  346.         change.at<float>(0, 0) = cvmGet(&ProVector, 7, 0);
  347.         change.at<float>(0, 1) = cvmGet(&ProVector, 7, 1);
  348.         change.at<float>(0, 2) = cvmGet(&ProVector, 7, 2);
  349.         change.at<float>(1, 0) = cvmGet(&ProVector, 7, 3);
  350.         change.at<float>(1, 1) = cvmGet(&ProVector, 7, 4);
  351.         change.at<float>(1, 2) = cvmGet(&ProVector, 7, 5);
  352.         change.at<float>(2, 0) = cvmGet(&ProVector, 7, 6);
  353.         change.at<float>(2, 1) = cvmGet(&ProVector, 7, 7);
  354.         change.at<float>(2, 2) = 1.f;
  356.         Mat temp1 = Mat::ones(1, N, CV_32FC1);
  357.         Mat temp2;                                        //存放处理后的特征点(x,y,1)T
  358.         temp2.create(3, N, CV_32FC1);
  360.         match2_xy_trans.copyTo(temp2(Range(0, 2), Range::all()));
  361.         temp1.copyTo(temp2(Range(2, 3), Range::all()));
  363.         Mat match2_xy_change = change * temp2;           //待配准图像中的特征点在参考图中的映射结果
  364.         Mat match2_xy_change_12 = match2_xy_change(Range(0, 2), Range::all());
  365.         //float* temp_ptr = match2_xy_change.ptr<float>(2);
  367.         float* temp_ptr = match2_xy_change.ptr<float>(2);
  369.         for (int i = 0; i < N; ++i)
  370.         {
  371.             float div_temp = temp_ptr[i];
  372.             match2_xy_change_12.at<float>(0, i) = match2_xy_change_12.at<float>(0, i) / div_temp;
  373.             match2_xy_change_12.at<float>(1, i) = match2_xy_change_12.at<float>(1, i) / div_temp;
  374.         }
  376.         Mat diff = match2_xy_change_12 - match1_xy_trans;
  377.         pow(diff, 2, diff);
  379.         rmse = (float)sqrt(sum(diff)(0) * 1.0 / N);      //sum输出是各个通道的和，rmse为输入点的均方误差
  380.     }
  381.     //如果是相似变换
  382.     else if (model == string("similarity"))
  383.     {
  384.         /*[x, y, 1, 0;
  385.           y, -x, 0, 1] * [a, b, c, d]'=[u,v]*/
  387.         Mat A3;
  388.         A3.create(2 * N, 4, CV_32FC1);
  389.         for (int i = 0; i < N; ++i)
  390.         {
  391.             A3.at<float>(2 * i, 0) = match2_xy.at<float>(i, 0);
  392.             A3.at<float>(2 * i, 1) = match2_xy.at<float>(i, 1);
  393.             A3.at<float>(2 * i, 2) = 1.f;
  394.             A3.at<float>(2 * i, 3) = 0.f;
  396.             A3.at<float>(2 * i + 1, 0) = match2_xy.at<float>(i, 1);
  397.             A3.at<float>(2 * i + 1, 1) = match2_xy.at<float>(i, 0) * (-1.f);
  398.             A3.at<float>(2 * i + 1, 2) = 0.f;
  399.             A3.at<float>(2 * i + 1, 3) = 1.f;
  400.         }
  402.         Vec4f values;
  403.         solve(A3, B, values, DECOMP_QR);
  404.         change = (Mat_<float>(3, 3) << values(0), values(1), values(2),
  405.             values(1) * (-1.0f), values(0), values(3),
  406.             +0.f, +0.f, 1.f);
  408.         Mat temp_1 = change(Range(0, 2), Range(0, 2));//尺度和旋转量
  409.         Mat temp_2 = change(Range(0, 2), Range(2, 3));//平移量
  411.         Mat match2_xy_change = temp_1 * match2_xy_trans + repeat(temp_2, 1, N);
  412.         Mat diff = match2_xy_change - match1_xy_trans;//求差
  413.         pow(diff, 2, diff);
  414.         rmse = (float)sqrt(sum(diff)(0) * 1.0) / N;//sum输出是各个通道的和
  415.     }
  417.     return change;
  418. }
  420. /*********************该函数删除错误的匹配点对****************************/
  421. /*points_1表示参考图像上匹配的特征点位置
  422.  points_2表示待配准图像上匹配的特征点位置
  423.  model表示变换模型，“similarity”,"affine"，“perspective”
  424.  threshold表示内点阈值
  425.  inliers表示points_1和points_2中对应的点对是否是正确匹配，如果是，对应元素值为1，否则为0
  426.  rmse表示最后所有正确匹配点对计算出来的误差
  427.  返回一个3 x 3矩阵，表示待配准图像到参考图像的变换矩阵
  428.  */
  429. Mat myMatch::ransac(const vector<Point2f>& points_1, const vector<Point2f>& points_2, string model, float threshold, vector<bool>& inliers, float& rmse)
  430. {
  431.     if (points_1.size() != points_2.size())
  432.         CV_Error(CV_StsBadArg, "ransac模块输入特征点数量不一致！");
  434.     if (!(model == string("affine") || model == string("similarity") ||
  435.         model == string("perspective") || model == string("projective")))
  436.         CV_Error(CV_StsBadArg, "ransac模块图像变换类型输入错误！");
  438.     const size_t N = points_1.size();                    //特征点对数，size_t 类型常用来保存一个数据的大小，通常为整形，可移植性强
  440.     int n;                                                //相当于不同模型对应的标签
  441.     size_t max_iteration, iterations;
  443.     //确定最大迭代次数，就目前来看此法过于简单粗暴，可以使用自适应迭代次数法
  444.     if (model == string("similarity"))
  445.     {
  446.         n = 2;
  447.         max_iteration = N * (N - 1) / 2;
  448.     }
  449.     else if (model == string("affine"))
  450.     {
  451.         n = 3;
  452.         max_iteration = N * (N - 1) * (N - 2) / (2 * 3);
  453.     }
  454.     else if (model == string("perspective"))
  455.     {
  456.         n = 4;
  457.         max_iteration = N * (N - 1) * (N - 2) / (2 * 3);
  458.     }
  460.     if (max_iteration > 800)
  461.         iterations = 800;
  462.     else
  463.         iterations = max_iteration;
  465.     //取出保存在points_1和points_2中的点坐标，保存在Mat矩阵中，方便处理
  466.     Mat arr_1, arr_2;                                    //arr_1,和arr_2是一个[3 x N]的矩阵，每一列表示一个点坐标,第三行全是1
  467.     arr_1.create(3, N, CV_32FC1);
  468.     arr_2.create(3, N, CV_32FC1);
  470.     //获取矩阵每一行的首地址
  471.     float* p10 = arr_1.ptr<float>(0), * p11 = arr_1.ptr<float>(1), * p12 = arr_1.ptr<float>(2);
  472.     float* p20 = arr_2.ptr<float>(0), * p21 = arr_2.ptr<float>(1), * p22 = arr_2.ptr<float>(2);
  474.     //把特征点放到矩阵中
  475.     for (size_t i = 0; i < N; ++i)
  476.     {
  477.         p10[i] = points_1[i].x;
  478.         p11[i] = points_1[i].y;
  479.         p12[i] = 1.f;
  481.         p20[i] = points_2[i].x;
  482.         p21[i] = points_2[i].y;
  483.         p22[i] = 1.f;
  484.     }
  486.     Mat rand_mat;                                //特征点索引
  487.     rand_mat.create(1, n, CV_32SC1);
  489.     int* p = rand_mat.ptr<int>(0);
  491.     Mat sub_arr1, sub_arr2;                        //存放随机挑选的特征点
  492.     sub_arr1.create(n, 2, CV_32FC1);
  493.     sub_arr2.create(n, 2, CV_32FC1);
  495.     Mat T;                                        //待配准图像到参考图像的变换矩阵
  496.     int most_consensus_num = 0;                    //当前最优一致集个数初始化为0
  498.     vector<bool> right;
  499.     right.resize(N);
  500.     inliers.resize(N);
  502.     for (size_t i = 0; i < iterations; ++i)        //迭代次数
  503.     {
  504.         //随机选择n个不同的点对，不同的模型每次随机选择的个数不同
  505.         while (1)
  506.         {
  507.             randu(rand_mat, 0, N - 1);            //随机生成n个范围在[0,N-1]之间的数，作为获取特征点的索引
  509.             //保证这n个点坐标不相同
  510.             if (n == 2 && p[0] != p[1] &&
  511.                 (p10[p[0]] != p10[p[1]] || p11[p[0]] != p11[p[1]]) &&
  512.                 (p20[p[0]] != p20[p[1]] || p21[p[0]] != p21[p[1]]))
  513.                 break;
  515.             if (n == 3 && p[0] != p[1] && p[0] != p[2] && p[1] != p[2] &&
  516.                 (p10[p[0]] != p10[p[1]] || p11[p[0]] != p11[p[1]]) &&
  517.                 (p10[p[0]] != p10[p[2]] || p11[p[0]] != p11[p[2]]) &&
  518.                 (p10[p[1]] != p10[p[2]] || p11[p[1]] != p11[p[2]]) &&
  519.                 (p20[p[0]] != p20[p[1]] || p21[p[0]] != p21[p[1]]) &&
  520.                 (p20[p[0]] != p20[p[2]] || p21[p[0]] != p21[p[2]]) &&
  521.                 (p20[p[1]] != p20[p[2]] || p21[p[1]] != p21[p[2]]))
  522.                 break;
  524.             if (n == 4 && p[0] != p[1] && p[0] != p[2] && p[0] != p[3] &&
  525.                 p[1] != p[2] && p[1] != p[3] && p[2] != p[3] &&
  526.                 (p10[p[0]] != p10[p[1]] || p11[p[0]] != p11[p[1]]) &&
  527.                 (p10[p[0]] != p10[p[2]] || p11[p[0]] != p11[p[2]]) &&
  528.                 (p10[p[0]] != p10[p[3]] || p11[p[0]] != p11[p[3]]) &&
  529.                 (p10[p[1]] != p10[p[2]] || p11[p[1]] != p11[p[2]]) &&
  530.                 (p10[p[1]] != p10[p[3]] || p11[p[1]] != p11[p[3]]) &&
  531.                 (p10[p[2]] != p10[p[3]] || p11[p[2]] != p11[p[3]]) &&
  532.                 (p20[p[0]] != p20[p[1]] || p21[p[0]] != p21[p[1]]) &&
  533.                 (p20[p[0]] != p20[p[2]] || p21[p[0]] != p21[p[2]]) &&
  534.                 (p20[p[0]] != p20[p[3]] || p21[p[0]] != p21[p[3]]) &&
  535.                 (p20[p[1]] != p20[p[2]] || p21[p[1]] != p21[p[2]]) &&
  536.                 (p20[p[1]] != p20[p[3]] || p21[p[1]] != p21[p[3]]) &&
  537.                 (p20[p[2]] != p20[p[3]] || p21[p[2]] != p21[p[3]]))
  538.                 break;
  539.         }
  541.         //提取出n个点对
  542.         for (int i = 0; i < n; ++i)
  543.         {
  544.             sub_arr1.at<float>(i, 0) = p10[p[i]];
  545.             sub_arr1.at<float>(i, 1) = p11[p[i]];
  547.             sub_arr2.at<float>(i, 0) = p20[p[i]];
  548.             sub_arr2.at<float>(i, 1) = p21[p[i]];
  549.         }
  551.         //根据这n个点对，计算初始变换矩阵 T
  552.         T = LMS(sub_arr1, sub_arr2, model, rmse);
  554.         int consensus_num = 0;                    //当前一致集(内点)个数
  556.         if (model == string("perspective"))
  557.         {
  558.             //变换矩阵计算待配准图像特征点在参考图像中的映射点
  559.             Mat match2_xy_change = T * arr_2;    //arr_2 中存放的是待配准图像的特征点 (3,N),
  561.             //match2_xy_change(Range(0, 2), Range::all())意思是提取 match2_xy_change 的 0、1 行，所有的列
  562.             Mat match2_xy_change_12 = match2_xy_change(Range(0, 2), Range::all());
  564.             //获取 match2_xy_change 第二行首地址
  565.             float* temp_ptr = match2_xy_change.ptr<float>(2);
  567.             for (size_t i = 0; i < N; ++i)
  568.             {
  569.                 float div_temp = temp_ptr[i];    //match2_xy_change 第二行第 i 列值，除以 div_temp ，是为了保证第三行为 1，和原始坐标相对应
  570.                 match2_xy_change_12.at<float>(0, i) = match2_xy_change_12.at<float>(0, i) / div_temp;
  571.                 match2_xy_change_12.at<float>(1, i) = match2_xy_change_12.at<float>(1, i) / div_temp;
  572.             }
  574.             //计算待配准图像特征点在参考图像中的映射点与参考图像中对应点的距离
  575.             Mat diff = match2_xy_change_12 - arr_1(Range(0, 2), Range::all());
  577.             pow(diff, 2, diff);
  579.             //第一行和第二行求和，即两点间距离的平方
  580.             Mat add = diff(Range(0, 1), Range::all()) + diff(Range(1, 2), Range::all());
  582.             float* p_add = add.ptr<float>(0);
  584.             //遍历所有距离，如果小于阈值，则认为是局内点
  585.             for (size_t i = 0; i < N; ++i)
  586.             {
  587.                 if (p_add[i] < threshold)                //初始 p_add[i]        
  588.                 {
  589.                     right[i] = true;
  590.                     ++consensus_num;
  591.                 }
  592.                 else
  593.                     right[i] = false;
  594.             }
  595.         }
  597.         else if (model == string("affine") || model == string("similarity"))
  598.         {
  599.             Mat match2_xy_change = T * arr_2;        //计算在参考图像中的映射坐标
  600.             Mat diff = match2_xy_change - arr_1;
  602.             pow(diff, 2, diff);
  604.             //第一行和第二行求和，计算特征点间的距离的平方
  605.             Mat add = diff(Range(0, 1), Range::all()) + diff(Range(1, 2), Range::all());
  607.             float* p_add = add.ptr<float>(0);
  609.             for (size_t i = 0; i < N; ++i)
  610.             {
  611.                 //如果小于阈值
  612.                 if (p_add[i] < threshold)
  613.                 {
  614.                     right[i] = true;
  615.                     ++consensus_num;
  616.                 }
  617.                 else
  618.                     right[i] = false;
  619.             }
  620.         }
  622.         //判断当前一致集是否是优于之前最优一致集，并更新当前最优一致集个数
  623.         if (consensus_num > most_consensus_num)
  624.         {
  625.             most_consensus_num = consensus_num;
  627.             //把正确匹配的点赋予标签 1 
  628.             for (size_t i = 0; i < N; ++i)
  629.                 inliers[i] = right[i];
  630.         }
  631.     }
  633.     //删除重复点对
  634.     for (size_t i = 0; i < N - 1; ++i)
  635.     {
  636.         for (size_t j = i + 1; j < N; ++j)
  637.         {
  638.             if (inliers[i] && inliers[j])
  639.             {
  640.                 if (p10[i] == p10[j] && p11[i] == p11[j] && p20[i] == p20[j] && p21[i] == p21[j])
  641.                 {
  642.                     inliers[j] = false;
  643.                     --most_consensus_num;
  644.                 }
  645.             }
  646.         }
  647.     }
  649.     //迭代结束，获得最优一致集合，根据这些最优一致集合计算出最终的变换关系 T
  650.     Mat consensus_arr1, consensus_arr2;                //经过迭代后最终确认正确匹配的点
  652.     consensus_arr1.create(most_consensus_num, 2, CV_32FC1);
  653.     consensus_arr2.create(most_consensus_num, 2, CV_32FC1);
  655.     int k = 0;
  657.     for (size_t i = 0; i < N; ++i)
  658.     {
  659.         if (inliers[i])
  660.         {
  661.             consensus_arr1.at<float>(k, 0) = p10[i];
  662.             consensus_arr1.at<float>(k, 1) = p11[i];
  664.             consensus_arr2.at<float>(k, 0) = p20[i];
  665.             consensus_arr2.at<float>(k, 1) = p21[i];
  666.             ++k;
  667.         }
  669.     }
  671.     int num_ransac = (model == string("similarity") ? 2 : (model == string("affine") ? 3 : 4));
  673.     if (k < num_ransac)
  674.         CV_Error(CV_StsBadArg, "ransac模块删除错误点对后剩下正确点对个数不足以计算出变换关系矩阵！");
  676.     //利用迭代后正确匹配点计算变换矩阵，为什么不是挑选 n 个点计算变换矩阵
  677.     T = LMS(consensus_arr1, consensus_arr2, model, rmse);
  679.     return T;
  680. }
  682. /********************该函数生成两幅图的棋盘网格图*************************/
  683. /*image_1表示参考图像
  684.  image_2表示配准后的待配准图像
  685.  chessboard_1表示image_1的棋盘图像
  686.  chessboard_2表示image_2的棋盘图像
  687.  mosaic_image表示image_1和image_2的镶嵌图像
  688.  width表示棋盘网格大小
  689.  */
  690. void myMatch::mosaic_map(const Mat& image_1, const Mat& image_2, Mat& chessboard_1, Mat& chessboard_2, Mat& mosaic_image, int width)
  691. {
  692.     if (image_1.size != image_2.size)
  693.         CV_Error(CV_StsBadArg, "mosaic_map模块输入两幅图大小必须一致！");
  695.     //生成image_1的棋盘网格图
  696.     chessboard_1 = image_1.clone();
  698.     int rows_1 = chessboard_1.rows;
  699.     int cols_1 = chessboard_1.cols;
  701.     int row_grids_1 = cvFloor((double)rows_1 / width);        //行方向网格个数
  702.     int col_grids_1 = cvFloor((double)cols_1 / width);        //列方向网格个数
  704.     //指定区域像素赋值为零，便形成了棋盘图
  706.     //第一幅图，第一行 2、4、6 像素值赋值零；第一幅图与第二幅图零像素位置交叉，以便两幅图交叉显示
  707.     for (int i = 0; i < row_grids_1; i = i + 2)
  708.     {
  709.         for (int j = 1; j < col_grids_1; j = j + 2)
  710.         {
  711.             Range range_x(j * width, (j + 1) * width);
  712.             Range range_y(i * width, (i + 1) * width);
  714.             chessboard_1(range_y, range_x) = 0;
  715.         }
  716.     }
  718.     for (int i = 1; i < row_grids_1; i = i + 2)
  719.     {
  720.         for (int j = 0; j < col_grids_1; j = j + 2)
  721.         {
  722.             Range range_x(j * width, (j + 1) * width);
  723.             Range range_y(i * width, (i + 1) * width);
  725.             chessboard_1(range_y, range_x) = 0;
  726.         }
  727.     }
  729.     //生成image_2的棋盘网格图
  731.     chessboard_2 = image_2.clone();
  733.     int rows_2 = chessboard_2.rows;
  734.     int cols_2 = chessboard_2.cols;
  736.     int row_grids_2 = cvFloor((double)rows_2 / width);//行方向网格个数
  737.     int col_grids_2 = cvFloor((double)cols_2 / width);//列方向网格个数
  739.     //第二幅图，第一行 1、3、5 像素值赋值零
  740.     for (int i = 0; i < row_grids_2; i = i + 2)
  741.     {
  742.         for (int j = 0; j < col_grids_2; j = j + 2)
  743.         {
  744.             Range range_x(j * width, (j + 1) * width);
  745.             Range range_y(i * width, (i + 1) * width);
  746.             chessboard_2(range_y, range_x) = 0;
  747.         }
  748.     }
  750.     for (int i = 1; i < row_grids_2; i = i + 2)
  751.     {
  752.         for (int j = 1; j < col_grids_2; j = j + 2)
  753.         {
  754.             Range range_x(j * width, (j + 1) * width);
  755.             Range range_y(i * width, (i + 1) * width);
  756.             chessboard_2(range_y, range_x) = 0;
  757.         }
  758.     }
  760.     //两个棋盘图进行叠加，显示配准效果
  761.     mosaic_image = chessboard_1 + chessboard_2;
  762. }
  764. /*该函数对输入图像指定位置像素进行中值滤波，消除边缘拼接阴影*/
  765. /*image表示输入的图像
  766.  position表示需要进行中值滤波的位置
  767.  */
  768. inline void median_filter(Mat& image, const vector<vector<int>>& pos)
  769. {
  770.     int channels = image.channels();
  771.     switch (channels)
  772.     {
  773.     case 1://单通道
  774.         for (auto beg = pos.cbegin(); beg != pos.cend(); ++beg)
  775.         {
  776.             int i = (*beg)[0];//y
  777.             int j = (*beg)[1];//x
  778.             uchar& pix_val = image.at<uchar>(i, j);
  779.             vector<uchar> pixs;
  780.             for (int row = -1; row <= 1; ++row)
  781.             {
  782.                 for (int col = -1; col <= 1; ++col)
  783.                 {
  784.                     if (i + row >= 0 && i + row < image.rows && j + col >= 0 && j + col < image.cols)
  785.                     {
  786.                         pixs.push_back(image.at<uchar>(i + row, j + col));
  787.                     }
  788.                 }
  789.             }
  790.             //排序
  791.             std::sort(pixs.begin(), pixs.end());
  792.             pix_val = pixs[pixs.size() / 2];
  793.         }
  794.         break;
  796.     case 3://3通道
  797.         for (auto beg = pos.cbegin(); beg != pos.cend(); ++beg)
  798.         {
  799.             int i = (*beg)[0];//y
  800.             int j = (*beg)[1];//x
  801.             Vec3b& pix_val = image.at<Vec3b>(i, j);
  802.             vector<cv::Vec3b> pixs;
  803.             for (int row = -1; row <= 1; ++row)
  804.             {
  805.                 for (int col = -1; col <= 1; ++col)
  806.                 {
  807.                     if (i + row >= 0 && i + row < image.rows && j + col >= 0 && j + col < image.cols)
  808.                     {
  809.                         pixs.push_back(image.at<Vec3b>(i + row, j + col));
  810.                     }
  811.                 }
  812.             }
  814.             //排序
  815.             std::sort(pixs.begin(), pixs.end(),
  816.                 [pix_val](const Vec3b& a, const Vec3b& b)->bool {
  817.                     return sum((a).ddot(a))[0] < sum((b).ddot(b))[0];
  818.                 });
  819.             pix_val = pixs[pixs.size() / 2];
  820.         }
  821.         break;
  822.     default:break;
  823.     }
  824. }
  826. /***************该函数把配准后的图像进行融合*****************/
  827. /*该函数功能主要是来对图像进行融合，以显示配准的效果
  828.  *image_1表示参考图像
  829.  *image_2表示待配准图像
  830.  *T表示待配准图像到参考图像的转换矩阵
  831.  *fusion_image表示参考图像和待配准图像融合后的图像
  832.  *mosaic_image表示参考图像和待配准图像融合镶嵌后的图像，镶嵌图形是为了观察匹配效果
  833.  *matched_image表示把待配准图像进行配准后的结果
  834.  */
  835. void myMatch::image_fusion(const Mat& image_1, const Mat& image_2, const Mat T, Mat& fusion_image, Mat& matched_image)
  836. {
  837.     //有关depth()的理解，详解：https://blog.csdn.net/datouniao1/article/details/113524784
  839.     if (!(image_1.depth() == CV_8U && image_2.depth() == CV_8U))
  840.         CV_Error(CV_StsBadArg, "image_fusion模块仅支持uchar类型图像！");
  841.     if (image_1.channels() == 4 || image_2.channels() == 4)
  842.         CV_Error(CV_StsBadArg, "image_fusion模块仅仅支持单通道或者3通道图像");
  844.     int rows_1 = image_1.rows, cols_1 = image_1.cols;
  845.     int rows_2 = image_2.rows, cols_2 = image_2.cols;
  847.     int channel_1 = image_1.channels();
  848.     int channel_2 = image_2.channels();
  850.     //可以对：彩色-彩色、彩色-灰色、灰色-彩色、灰色-灰色的配准
  851.     Mat image_1_temp, image_2_temp;
  853.     if (channel_1 == 3 && channel_2 == 3)
  854.     {
  855.         image_1_temp = image_1;
  856.         image_2_temp = image_2;
  857.     }
  858.     else if (channel_1 == 1 && channel_2 == 3)
  859.     {
  860.         image_1_temp = image_1;
  861.         cvtColor(image_2, image_2_temp, CV_RGB2GRAY);    //颜色空间转换，把彩色图转化为灰度图
  862.     }
  863.     else if (channel_1 == 3 && channel_2 == 1)
  864.     {
  865.         cvtColor(image_1, image_1_temp, CV_RGB2GRAY);
  866.         image_2_temp = image_2;
  867.     }
  868.     else if (channel_1 == 1 && channel_2 == 1)
  869.     {
  870.         image_1_temp = image_1;
  871.         image_2_temp = image_2;
  872.     }
  874.     //创建一个（3，3）float 矩阵 Mat_ 是一个模板类
  875.     Mat T_temp = (Mat_<float>(3, 3) << 1, 0, cols_1, 0, 1, rows_1, 0, 0, 1);
  876.     Mat T_1 = T_temp * T;
  878.     //对参考图像和待配准图像进行变换
  879.     Mat trans_1, trans_2;//same type as image_2_temp 
  881.     trans_1 = Mat::zeros(3 * rows_1, 3 * cols_1, image_1_temp.type());                    //创建扩大后的矩阵
  882.     image_1_temp.copyTo(trans_1(Range(rows_1, 2 * rows_1), Range(cols_1, 2 * cols_1)));    //把image_1_temp中的数据复制到扩大后矩阵的对应位置
  884.     warpPerspective(image_2_temp, trans_2, T_1, Size(3 * cols_1, 3 * rows_1), INTER_LINEAR, 0, Scalar::all(0));
  886.     /*功能：把image_2_temp投射到一个新的视平面，即变形
  887.      *image_2_temp为输入矩阵
  888.      *trans_2为输出矩阵，尺寸和输入矩阵大小一致
  889.      *T_1为变换矩阵，(3, 3)矩阵用于透视变换, (2, 2)用于线性变换, (1, 1)用于平移
  890.      *warpPerspective函数功能是进行透视变换，T_1是(3,3)的透视变换矩阵
  891.      *int flags=INTER_LINEAR为输出图像的插值方法
  892.      *int borderMode=BORDER_CONSTANT，0 为图像边界的填充方式
  893.      *const Scalar& borderValue=Scalar()：边界的颜色设置，一般默认是0，Scalar::all(0)对影像边界外进行填充*/
  895.      //使用简单的均值法进行图像融合
  896.     Mat trans = trans_2.clone();                            //把经过透视变换的image_2复制给trans
  897.     int nRows = rows_1;
  898.     int nCols = cols_1;
  899.     int len = nCols;
  900.     bool flag_1 = false;
  901.     bool flag_2 = false;
  903.     vector<vector<int>> positions;                            //保存边缘位置坐标
  905.     switch (image_1_temp.channels())
  906.     {
  907.     case 1:                                                    //如果图像1是单通道的
  908.         for (int i = 0; i < nRows; ++i)
  909.         {
  910.             uchar* ptr_1 = trans_1.ptr<uchar>(i + rows_1);        //访问trans_1中的指定行像素值
  911.             uchar* ptr = trans.ptr<uchar>(i + rows_1);            //访问trans  中的指定行像素值
  913.             for (int j = 0; j < nCols; ++j)
  914.             {
  915.                 if (ptr[j + len] == 0 && ptr_1[j + len] != 0)    //非重合区域
  916.                 {
  917.                     flag_1 = true;
  918.                     if (flag_2)                                    //表明从重合区域过度到了非重合区域
  919.                     {
  920.                         for (int p = -1; p <= 1; ++p)            //保存边界3x3区域像素
  921.                         {
  922.                             for (int q = -1; q <= 1; ++q)
  923.                             {
  924.                                 vector<int> pos;
  925.                                 pos.push_back(i + rows_1 + p);
  926.                                 pos.push_back(j + cols_1 + q);
  927.                                 positions.push_back(pos);//保存边缘位置坐标
  928.                             }
  929.                         }
  930.                         flag_2 = false;
  931.                     }
  932.                     ptr[j + len] = ptr_1[j + len];
  933.                 }
  934.                 else//对于重合区域
  935.                 {
  936.                     flag_2 = true;
  937.                     if (flag_1)//表明从非重合区域过度到了重合区域
  938.                     {
  939.                         for (int p = -1; p <= 1; ++p)//保存边界3x3区域像素
  940.                         {
  941.                             for (int q = -1; q <= 1; ++q)
  942.                             {
  943.                                 vector<int> pos;
  944.                                 pos.push_back(i + rows_1 + p);
  945.                                 pos.push_back(j + cols_1 + q);
  946.                                 positions.push_back(pos);//保存边缘位置坐标
  947.                             }
  948.                         }
  949.                         flag_1 = false;
  950.                     }
  951.                     ptr[j + len] = saturate_cast<uchar>(((float)ptr[j + len] + (float)ptr_1[j + len]) / 2);
  952.                 }
  953.             }
  954.         }
  955.         break;
  956.     case 3:                                                    //如果图像是三通道的
  957.         len = len * image_1_temp.channels();                //3倍的列数
  958.         for (int i = 0; i < nRows; ++i)
  959.         {
  960.             uchar* ptr_1 = trans_1.ptr<uchar>(i + rows_1);    //访问trans_1中的指定行像素值
  961.             uchar* ptr = trans.ptr<uchar>(i + rows_1);        //访问trans  中的指定行像素值
  963.             for (int j = 0; j < nCols; ++j)
  964.             {
  965.                 int nj = j * image_1_temp.channels();
  967.                 //若两张影像对应列像素值不同（3通道），则是非重合区，该过程仅仅是为了使配准后的影像进行融合，而非配准
  968.                 if (ptr[nj + len] == 0 && ptr[nj + len + 1] == 0 && ptr[nj + len + 2] == 0 &&
  969.                     ptr_1[nj + len] != 0 && ptr_1[nj + len + 1] != 0 && ptr_1[nj + len + 2] != 0)
  970.                 {
  971.                     flag_1 = true;
  972.                     if (flag_2)                                //表明从重合区域过度到了非重合区域
  973.                     {
  974.                         for (int p = -1; p <= 1; ++p)        //保存边界3x3区域像素
  975.                         {
  976.                             for (int q = -1; q <= 1; ++q)
  977.                             {
  978.                                 vector<int> pos;
  979.                                 pos.push_back(i + rows_1 + p);
  980.                                 pos.push_back(j + cols_1 + q);
  981.                                 positions.push_back(pos);    //保存边缘位置坐标
  982.                             }
  983.                         }
  984.                         flag_2 = false;
  985.                     }
  986.                     ptr[nj + len] = ptr_1[nj + len];
  987.                     ptr[nj + len + 1] = ptr_1[nj + len + 1];
  988.                     ptr[nj + len + 2] = ptr_1[nj + len + 2];
  989.                 }
  990.                 else
  991.                 {                                            //对于重合区域
  992.                     flag_2 = true;
  993.                     if (flag_1)                                //表明从非重合区域过度到了重合区域
  994.                     {
  995.                         for (int p = -1; p <= 1; ++p)        //保存边界3x3区域像素
  996.                         {
  997.                             for (int q = -1; q <= 1; ++q)
  998.                             {
  999.                                 vector<int> pos;
  1000.                                 pos.push_back(i + rows_1 + p);
  1001.                                 pos.push_back(j + cols_1 + q);
  1002.                                 positions.push_back(pos);    //保存边缘位置坐标
  1003.                             }
  1004.                         }
  1005.                         flag_1 = false;
  1006.                     }
  1007.                     ptr[nj + len] = saturate_cast<uchar>(((float)ptr[nj + len] + (float)ptr_1[nj + len]) / 2);
  1008.                     ptr[nj + len + 1] = saturate_cast<uchar>(((float)ptr[nj + len + 1] + (float)ptr_1[nj + len + 1]) / 2);
  1009.                     ptr[nj + len + 2] = saturate_cast<uchar>(((float)ptr[nj + len + 2] + (float)ptr_1[nj + len + 2]) / 2);
  1010.                 }
  1011.             }
  1012.         }
  1013.         break;
  1014.     default:break;
  1015.     }
  1017.     //根据获取的边缘区域的坐标，对边缘像素进行中值滤波，消除边缘效应
  1018.     median_filter(trans, positions);
  1020.     //删除多余的区域
  1021.     Mat left_up = T_1 * (Mat_<float>(3, 1) << 0, 0, 1);                            //左上角
  1022.     Mat left_down = T_1 * (Mat_<float>(3, 1) << 0, rows_2 - 1, 1);                //左下角
  1023.     Mat right_up = T_1 * (Mat_<float>(3, 1) << cols_2 - 1, 0, 1);                //右上角
  1024.     Mat right_down = T_1 * (Mat_<float>(3, 1) << cols_2 - 1, rows_2 - 1, 1);    //右下角
  1026.     //对于透视变换，需要除以一个因子
  1027.     left_up = left_up / left_up.at<float>(2, 0);
  1028.     left_down = left_down / left_down.at<float>(2, 0);
  1029.     right_up = right_up / right_up.at<float>(2, 0);
  1030.     right_down = right_down / right_down.at<float>(2, 0);
  1032.     //计算x,y坐标的范围
  1033.     float temp_1 = min(left_up.at<float>(0, 0), left_down.at<float>(0, 0));
  1034.     float temp_2 = min(right_up.at<float>(0, 0), right_down.at<float>(0, 0));
  1035.     float min_x = min(temp_1, temp_2);
  1037.     temp_1 = max(left_up.at<float>(0, 0), left_down.at<float>(0, 0));
  1038.     temp_2 = max(right_up.at<float>(0, 0), right_down.at<float>(0, 0));
  1039.     float max_x = max(temp_1, temp_2);
  1041.     temp_1 = min(left_up.at<float>(1, 0), left_down.at<float>(1, 0));
  1042.     temp_2 = min(right_up.at<float>(1, 0), right_down.at<float>(1, 0));
  1043.     float min_y = min(temp_1, temp_2);
  1045.     temp_1 = max(left_up.at<float>(1, 0), left_down.at<float>(1, 0));
  1046.     temp_2 = max(right_up.at<float>(1, 0), right_down.at<float>(1, 0));
  1047.     float max_y = max(temp_1, temp_2);
  1049.     int X_min = max(cvFloor(min_x), 0);
  1050.     int X_max = min(cvCeil(max_x), 3 * cols_1 - 1);
  1051.     int Y_min = max(cvFloor(min_y), 0);
  1052.     int Y_max = min(cvCeil(max_y), 3 * rows_1 - 1);
  1054.     if (X_min > cols_1)
  1055.         X_min = cols_1;
  1056.     if (X_max < 2 * cols_1 - 1)
  1057.         X_max = 2 * cols_1 - 1;
  1058.     if (Y_min > rows_1)
  1059.         Y_min = rows_1;
  1060.     if (Y_max < 2 * rows_1 - 1)
  1061.         Y_max = 2 * rows_1 - 1;
  1063.     //提取有价值区域
  1064.     Range Y_range(Y_min, Y_max + 1);
  1065.     Range X_range(X_min, X_max + 1);
  1067.     fusion_image = trans(Y_range, X_range);
  1068.     matched_image = trans_2(Y_range, X_range);
  1070.     Mat ref_matched = trans_1(Y_range, X_range);
  1072.     //生成棋盘网格图像
  1073.     /*Mat chessboard_1, chessboard_2;
  1075.     mosaic_map(trans_1(Y_range, X_range), trans_2(Y_range, X_range), chessboard_1, chessboard_2, mosaic_image, 100);*/
  1077.     /*cv::imwrite(".\\image_save\\参考图像棋盘图像.jpg", chessboard_1);
  1078.     cv::imwrite(".\\image_save\\待配准图像棋盘图像.jpg", chessboard_2);*/
  1079.     cv::imwrite(".\\image_save\\配准后的参考图像.jpg", ref_matched);
  1080.     cv::imwrite(".\\image_save\\配准后的待配准图像.jpg", matched_image);
  1081. }
  1083.  /******该函数计算参考图像一个描述子和待配准图像所有描述子的欧式距离，并获得最近邻和次近邻距离，以及对应的索引*/
  1084.  /*sub_des_1表示参考图像的一个描述子
  1085.   *des_2表示待配准图像描述子
  1086.   *num_des_2值待配准图像描述子个数
  1087.   *dims_des指的是描述子维度
  1088.   *dis保存最近邻和次近邻距离
  1089.   *idx保存最近邻和次近邻索引
  1090.   */
  1091. inline void min_dis_idx(const float* ptr_1, const Mat& des_2, int num_des2, int dims_des, float dis[2], int idx[2])
  1092. {
  1093.     float min_dis1 = 1000, min_dis2 = 2000;
  1094.     int min_idx1, min_idx2;
  1096.     for (int j = 0; j < num_des2; ++j)
  1097.     {
  1098.         const float* ptr_des_2 = des_2.ptr<float>(j);
  1099.         float cur_dis = 0;
  1100.         for (int k = 0; k < dims_des; ++k)
  1101.         {
  1102.             float diff = ptr_1[k] - ptr_des_2[k];
  1103.             cur_dis += diff * diff;
  1104.         }
  1105.         if (cur_dis < min_dis1) {
  1106.             min_dis1 = cur_dis;
  1107.             min_idx1 = j;
  1108.         }
  1109.         else if (cur_dis >= min_dis1 && cur_dis < min_dis2) {
  1110.             min_dis2 = cur_dis;
  1111.             min_idx2 = j;
  1112.         }
  1114.     }
  1115.     dis[0] = sqrt(min_dis1); dis[1] = sqrt(min_dis2);
  1116.     idx[0] = min_idx1; idx[1] = min_idx2;
  1117. }
  1119. /*加速版本的描述子匹配，返回匹配点候选集函数,该加速版本比前面版本速度提升了3倍*/
  1120. void myMatch::match_des(const Mat& des_1, const Mat& des_2, vector<vector<DMatch>>& dmatchs, DIS_CRIT dis_crite)
  1121. {
  1122.     int num_des_1 = des_1.rows;
  1123.     int num_des_2 = des_2.rows;
  1124.     int dims_des = des_1.cols;
  1126.     vector<DMatch> match(2);
  1128.     //对于参考图像上的每一点，和待配准图像进行匹配
  1129.     if (dis_crite == 0)                                    //欧几里得距离
  1130.     {
  1131.         for (int i = 0; i < num_des_1; ++i)                //对于参考图像中的每个描述子
  1132.         {
  1133.             const float* ptr_des_1 = des_1.ptr<float>(i);
  1134.             float dis[2];
  1135.             int idx[2];
  1136.             min_dis_idx(ptr_des_1, des_2, num_des_2, dims_des, dis, idx);
  1137.             match[0].queryIdx = i;
  1138.             match[0].trainIdx = idx[0];
  1139.             match[0].distance = dis[0];
  1141.             match[1].queryIdx = i;
  1142.             match[1].trainIdx = idx[1];
  1143.             match[1].distance = dis[1];
  1145.             dmatchs.push_back(match);
  1146.         }
  1147.     }
  1148.     else if (dis_crite == 1)//cos距离
  1149.     {
  1150.         Mat trans_des2;
  1151.         transpose(des_2, trans_des2);
  1152.         double aa = (double)getTickCount();
  1153.         Mat mul_des = des_1 * trans_des2;
  1154.         //gemm(des_1, des_2, 1, Mat(), 0, mul_des, GEMM_2_T);
  1155.         double time1 = ((double)getTickCount() - aa) / getTickFrequency();
  1156.         cout << "cos距离中矩阵乘法花费时间： " << time1 << "s" << endl;
  1158.         for (int i = 0; i < num_des_1; ++i)
  1159.         {
  1160.             float max_cos1 = -1000, max_cos2 = -2000;
  1161.             int max_idx1, max_idx2;
  1163.             float* ptr_1 = mul_des.ptr<float>(i);
  1164.             for (int j = 0; j < num_des_2; ++j)
  1165.             {
  1166.                 float cur_cos = ptr_1[j];
  1167.                 if (cur_cos > max_cos1) {
  1168.                     max_cos1 = cur_cos;
  1169.                     max_idx1 = j;
  1170.                 }
  1171.                 else if (cur_cos <= max_cos1 && cur_cos > max_cos2) {
  1172.                     max_cos2 = cur_cos;
  1173.                     max_idx2 = j;
  1174.                 }
  1175.             }
  1177.             match[0].queryIdx = i;
  1178.             match[0].trainIdx = max_idx1;
  1179.             match[0].distance = acosf(max_cos1);
  1181.             match[1].queryIdx = i;
  1182.             match[1].trainIdx = max_idx2;
  1183.             match[1].distance = acosf(max_cos2);
  1184.             dmatchs.push_back(match);
  1185.         }
  1186.     }
  1187. }
  1189. /*******************建立尺度直方图、ROM 直方图************************/
  1190. void myMatch::scale_ROM_Histogram(const vector<DMatch>& matches, float* scale_hist, float* ROM_hist, int n)
  1191. {
  1192.     int len = matches.size();
  1194.     //使用AutoBuffer分配一段内存，这里多出4个空间的目的是为了方便后面平滑直方图的需要
  1195.     AutoBuffer<float> buffer((4 * len) + n + 4);
  1197.     //X保存水平差分，Y保存数值差分，Mag保存梯度幅度，Ori保存梯度角度，W保存高斯权重
  1198.     float* X = buffer, * Y = buffer + len, * Mag = Y, * Ori = Y + len, * W = Ori + len;
  1199.     float* temp_hist = W + len + 2;                        //临时保存直方图数据
  1201.     for (int i = 0; i < n; ++i)
  1202.         temp_hist[i] = 0.f;                                //数据清零
  1203. }
  1205. /*******************该函数删除错误匹配点对,并完成匹配************************/
  1206. /*image_1表示参考图像，
  1207.   image_2表示待配准图像
  1208.   dmatchs表示最近邻和次近邻匹配点对
  1209.   keys_1表示参考图像特征点集合
  1210.   keys_2表示待配准图像特征点集合
  1211.   model表示变换模型
  1212.   right_matchs表示参考图像和待配准图像正确匹配点对
  1213.   matched_line表示在参考图像和待配准图像上绘制连接线
  1214.   该函数返回变换模型参数
  1215.  */
  1216. Mat myMatch::match(const Mat& image_1, const Mat& image_2, const vector<vector<DMatch>>& dmatchs, vector<KeyPoint> keys_1,
  1217.     vector<KeyPoint> keys_2, string model, vector<DMatch>& right_matchs, Mat& matched_line, vector<DMatch>& init_matchs)
  1218. {
  1219.     //获取初始匹配的关键点的位置
  1220.     vector<Point2f> point_1, point_2;
  1222.     for (size_t i = 0; i < dmatchs.size(); ++i)                        //增加一个一个0.8，正确匹配点数增加2
  1223.     {
  1224.         double dis_1 = dmatchs[i][0].distance;                        //distance对应的是特征点描述符的欧式距离
  1225.         double dis_2 = dmatchs[i][1].distance;
  1227.         //比率测试筛选误匹配点(初步筛选)，如果满足则认为是有候选集中有正确匹配点
  1228.         if ((dis_1 / dis_2) < dis_ratio3)                            //最近邻和次近邻距离比阈值
  1229.         {
  1230.             //queryIdx、trainIdx和distance是DMatch类中的一些属性    //pt是KeyPoint类中的成员，对应关键点的坐标
  1231.             point_1.push_back(keys_1[dmatchs[i][0].queryIdx].pt);    //queryIdx对应的是特征描述子的下标，也是对应特征点的下标
  1232.             point_2.push_back(keys_2[dmatchs[i][0].trainIdx].pt);    //trainIdx对应的是特征描述子的下标，也是对应特征点的下标
  1233.             init_matchs.push_back(dmatchs[i][0]);                    //保存正确的dmatchs
  1234.         }
  1235.     }
  1237.     cout << "距离比之后初始匹配点对个数是： " << init_matchs.size() << endl;
  1239.     int min_pairs = (model == string("similarity") ? 2 : (model == string("affine") ? 3 : 4));
  1241.     if (point_1.size() < min_pairs)
  1242.         CV_Error(CV_StsBadArg, "match模块距离比阶段匹配特征点个数不足！");
  1244.     //使用ransac算法再次对匹配点进行筛选，然后使用最后确定的匹配点计算变换矩阵的参数
  1245.     vector<bool> inliers;                                            //存放的是 bool 类型的数据，对应特征点
  1246.     float rmse;
  1248.     //homography是一个(3,3)矩阵，是待配准影像到参考影像的变换矩阵，初始误差阈值是 1.5
  1249.     Mat homography = ransac(point_1, point_2, model, 1.5, inliers, rmse);
  1251.     //提取出处正确匹配点对
  1252.     int i = 0;
  1253.     vector<Point2f> point_11, point_22;
  1254.     vector<DMatch>::iterator itt = init_matchs.begin();
  1255.     for (vector<bool>::iterator it = inliers.begin(); it != inliers.end(); ++it, ++itt)
  1256.     {
  1257.         if (*it)                                //如果是正确匹配点对
  1258.         {
  1259.             right_matchs.push_back(*itt);
  1261.             // init_matchs 中匹配点对的存储顺序和 point_1 中特征点的存储顺序是一一对应的
  1262.             point_11.push_back(point_1.at(i));
  1263.             point_22.push_back(point_2.at(i));
  1264.         }
  1265.         ++i;
  1267.     }
  1268.     cout << "使用RANSAC删除错误点对,且返回正确匹配个数： " << right_matchs.size() << endl;
  1269.     cout << "误差rmse: " << rmse << endl;
  1271.     //绘制初始匹配点对连线图,此时初始匹配指的是经过 KnnMatch 筛选后的匹配
  1272.     Mat initial_matched;                                                    //输出矩阵，类似画布
  1273.     drawMatches(image_1, keys_1, image_2, keys_2, init_matchs, initial_matched,
  1274.         Scalar(255, 0, 255), Scalar(0, 255, 0), vector<char>());    //该函数用于绘制特征点并对匹配的特征点进行连线
  1275.     imwrite(".\\image_save\\初始匹配点对.jpg", initial_matched);            //保存图片,第一个颜色控制连线，第二个颜色控制特征点
  1277.     //绘制正确匹配点对连线图
  1278.     drawMatches(image_1, keys_1, image_2, keys_2, right_matchs, matched_line,
  1279.         Scalar(255, 0, 255), Scalar(0, 255, 0), vector<char>());
  1280.     imwrite(".\\image_save\\正确匹配点对.jpg", matched_line);
  1282.     //保存和显示检测到的特征点
  1283.     Mat keys_image_1, keys_image_2;                                            //输出矩阵，类似画布
  1284.     drawKeypoints(image_1, keys_1, keys_image_1, Scalar(0, 255, 0));        //该函数用于绘制图像中的特征点
  1285.     drawKeypoints(image_2, keys_2, keys_image_2, Scalar(0, 255, 0));
  1286.     imwrite(".\\image_save\\参考图像检测到的特征点.jpg", keys_image_1);
  1287.     imwrite(".\\image_save\\待配准图像检测到的.jpg", keys_image_2);
  1289.     return homography;
  1290. }
  1292. myMatch::~myMatch()
  1293. {
  1294. }

5、 myDisplay.h

  1. #pragma
  3. #include<iostream>
  4. #include<opencv2\core\core.hpp>
  5. #include<opencv2\features2d\features2d.hpp>
  6. #include<vector>
  8. using namespace cv;
  9. using namespace std;
  11. class myDisplay
  12. {
  13. public:
  14.     myDisplay() {}
  15.     void mosaic_pyramid(const vector<vector<Mat>>& pyramid, Mat& pyramid_image, int nOctaceLayers, string str);
  17.     void write_mosaic_pyramid(const vector<vector<Mat>>& gauss_pyr_1, const vector<vector<Mat>>& dog_pyr_1,
  18.         const vector<vector<Mat>>& gauss_pyr_2, const vector<vector<Mat>>& dog_pyr_2, int nOctaveLayers);
  20.     //没用到，后文没有对其进行定义
  21.     void write_keys_image(vector<KeyPoint>& keypoints_1, vector<KeyPoint>& keypoints_2,
  22.         const Mat& image_1, const Mat& image_2, Mat& image_1_keys, Mat& image_2_keys, bool double_size);
  23. };

6、 myDisplay.cpp

  1. #include "myDisplay.h"
  3. #include<opencv2\highgui\highgui.hpp>
  4. #include<vector>
  5. #include<sstream>
  7. /***************************该函数把金字塔放在一副图像上(包络高斯金字塔和DOG金字塔)******************/
  8. /*pyramid表示高斯金字塔或者DOG金字塔
  9.  pyramid_image表示生成的金字塔图像
  10.  nOctaveLayers表示每组中间层数，默认是3
  11.  str表示是高斯金字塔还是DOG金字塔
  12. */
  13. void  myDisplay::mosaic_pyramid(const vector<vector<Mat>>& pyramid, Mat& pyramid_image, int nOctaceLayers, string str)
  14. {
  15.     //获得每组金字塔图像大小
  16.     vector<vector<int>> temp_size;
  17.     for (auto beg = pyramid.cbegin(); beg != pyramid.cend(); ++beg)
  18.     {
  19.         vector<int> temp_vec;
  20.         int rows = (*beg)[0].rows;
  21.         int cols = (*beg)[0].cols;
  22.         temp_vec.push_back(rows);
  23.         temp_vec.push_back(cols);
  24.         temp_size.push_back(temp_vec);
  25.     }
  27.     //计算最后生成的金字塔图像pyramid_image的大小
  28.     int total_rows = 0, total_cols = 0;
  29.     for (auto beg = temp_size.begin(); beg != temp_size.end(); ++beg)
  30.     {
  31.         total_rows = total_rows + (*beg)[0];//获取行大小
  32.         if (beg == temp_size.begin()) {
  33.             if (str == string("高斯金字塔"))
  34.                 total_cols = (nOctaceLayers + 3) * ((*beg)[1]);//获取列大小
  35.             else if (str == string("DOG金字塔"))
  36.                 total_cols = (nOctaceLayers + 2) * ((*beg)[1]);//获取列大小
  37.         }
  38.     }
  40.     pyramid_image.create(total_rows, total_cols, CV_8UC1);
  41.     int i = 0, accumulate_row = 0;
  42.     for (auto beg = pyramid.cbegin(); beg != pyramid.cend(); ++beg, ++i)
  43.     {
  44.         int accumulate_col = 0;
  45.         accumulate_row = accumulate_row + temp_size[i][0];
  46.         for (auto it = (*beg).cbegin(); it != (*beg).cend(); ++it)
  47.         {
  48.             accumulate_col = accumulate_col + temp_size[i][1];
  49.             Mat temp(pyramid_image, Rect(accumulate_col - temp_size[i][1], accumulate_row - temp_size[i][0], it->cols, it->rows));
  50.             Mat temp_it;
  51.             normalize(*it, temp_it, 0, 255, NORM_MINMAX, CV_32FC1);
  52.             convertScaleAbs(temp_it, temp_it, 1, 0);
  53.             temp_it.copyTo(temp);
  54.         }
  55.     }
  56. }
  58. /**************************该函数保存拼接后的高斯金字塔和DOG金字塔图像**************************/
  59. /*gauss_pyr_1表示参考高斯金字塔
  60.  dog_pyr_1表示参考DOG金字塔
  61.  gauss_pyr_2表示待配准高斯金字塔
  62.  dog_pyr_2表示待配准DOG金字塔
  63.  nOctaveLayers表示金字塔每组中间层数
  64.  */
  65. void myDisplay::write_mosaic_pyramid(const vector<vector<Mat>>& gauss_pyr_1, const vector<vector<Mat>>& dog_pyr_1,
  66.     const vector<vector<Mat>>& gauss_pyr_2, const vector<vector<Mat>>& dog_pyr_2, int nOctaveLayers)
  67. {
  69.     //显示参考和待配准高斯金字塔图像
  70.     Mat gauss_image_1, gauss_image_2;
  71.     mosaic_pyramid(gauss_pyr_1, gauss_image_1, nOctaveLayers, string("高斯金字塔"));
  72.     mosaic_pyramid(gauss_pyr_2, gauss_image_2, nOctaveLayers, string("高斯金字塔"));
  73.     imwrite(".\\image_save\\参考图像高斯金字塔.jpg", gauss_image_1);
  74.     imwrite(".\\image_save\\待配准图像高斯金字塔.jpg", gauss_image_2);
  76.     //显示参考和待配准DOG金字塔图像
  77.     Mat dog_image_1, dog_image_2;
  78.     mosaic_pyramid(dog_pyr_1, dog_image_1, nOctaveLayers, string("DOG金字塔"));
  79.     mosaic_pyramid(dog_pyr_2, dog_image_2, nOctaveLayers, string("DOG金字塔"));
  80.     imwrite(".\\image_save\\参考图像DOG金字塔.jpg", dog_image_1);
  81.     imwrite(".\\image_save\\待配准图像DOG金字塔.jpg", dog_image_2);
  82. }

7、 main.cpp

  1. #include"mySift.h"
  2. #include"myDisplay.h"
  3. #include"myMatch.h"
  4. #include<direct.h>
  6. #include<opencv2\highgui\highgui.hpp>
  7. #include<opencv2\calib3d\calib3d.hpp>
  8. #include<opencv2\imgproc\imgproc.hpp>
  10. #include<fstream>
  11. #include<stdlib.h>
  12. #include<direct.h>
  14. int main(int argc, char* argv[])
  15. {
  16.     /********************** 1、读入数据 **********************/
  17.     Mat image_1 = imread("..\\test_data\\perspective_graf_1.ppm", -1);
  18.     Mat image_2 = imread("..\\test_data\\perspective_graf_2.ppm", -1);
  20.     string change_model = "perspective";                            //affine为仿射变换，初始为perspective
  22.     string folderPath = ".\\image_save";
  23.     _mkdir(folderPath.c_str());
  25.     double total_count_beg = (double)getTickCount();                //算法运行总时间开始计时
  27.     mySift sift_1(0, 3, 0.04, 10, 1.6, true);                        //类对象
  29.     /********************** 1、参考图像特征点检测和描述 **********************/
  31.     vector<vector<Mat>> gauss_pyr_1, dog_pyr_1;                        //高斯金字塔和高斯差分金字塔
  32.     vector<KeyPoint> keypoints_1;                                    //特征点
  33.     vector<vector<vector<float>>> all_cell_contrasts_1;                //所有尺度层中所有单元格的对比度
  34.     vector<vector<float>> average_contrast_1;                        //所有尺度层中多有单元格的平均对比度
  35.     vector<vector<int>> n_cells_1;                                    //所有尺度层，每一尺度层中所有单元格所需特征数
  36.     vector<int> num_cell_1;                                            //当前尺度层中所有单元格中的所需特征数
  37.     vector<vector<int>> available_n_1;                                //所有尺度层，每一尺度层中所有单元格可得到特征数
  38.     vector<int> available_num_1;                                    //一个尺度层中所有单元格中可用特征数量
  39.     vector<KeyPoint> final_keypoints1;                                //第一次筛选结果
  40.     vector<KeyPoint> Final_keypoints1;                                //第二次筛选结果
  41.     vector<KeyPoint> Final_Keypoints1;                                //第三次筛选结果
  43.     Mat descriptors_1;                                                //特征描述子
  45.     double detect_1 = (double)getTickCount();                        //特征点检测运行时间开始计时
  47.     sift_1.detect(image_1, gauss_pyr_1, dog_pyr_1, keypoints_1, all_cell_contrasts_1,
  48.         average_contrast_1, n_cells_1, num_cell_1, available_n_1, available_num_1, final_keypoints1, Final_keypoints1, Final_Keypoints1);            //特征点检测
  50.     cout << "参考图像检测出的总特征数 =" << keypoints_1.size() << endl;
  52.     //getTickFrequency() 返回值为一秒的计时周期数，二者比值为特征点检测时间
  53.     double detect_time_1 = ((double)getTickCount() - detect_1) / getTickFrequency();
  55.     cout << "参考图像特征点检测时间是： " << detect_time_1 << "s" << endl;
  57.     double comput_1 = (double)getTickCount();
  59.     vector<Mat> sar_harris_fun_1;
  60.     vector<Mat> amplit_1;
  61.     vector<Mat> orient_1;
  63.     sift_1.comput_des(gauss_pyr_1, keypoints_1, amplit_1, orient_1, descriptors_1);
  65.     double comput_time_1 = ((double)getTickCount() - comput_1) / getTickFrequency();
  67.     cout << "参考图像特征点描述时间是： " << comput_time_1 << "s" << endl;
  69.     /********************** 1、待配准图像特征点检测和描述 **********************/
  71.     vector<vector<Mat>> gauss_pyr_2, dog_pyr_2;
  72.     vector<KeyPoint> keypoints_2;
  73.     vector<vector<vector<float>>> all_cell_contrasts_2;                //所有尺度层中所有单元格的对比度
  74.     vector<vector<float>> average_contrast_2;                        //所有尺度层中多有单元格的平均对比度
  75.     vector<vector<int>> n_cells_2;                                    //所有尺度层，每一尺度层中所有单元格所需特征数
  76.     vector<int> num_cell_2;                                            //当前尺度层中所有单元格中的所需特征数
  77.     vector<vector<int>> available_n_2;                                //所有尺度层，每一尺度层中的一个单元格可得到特征数
  78.     vector<int> available_num_2;                                    //一个尺度层中所有单元格中可用特征数量
  79.     vector<KeyPoint> final_keypoints2;                                //第一次筛选结果
  80.     vector<KeyPoint> Final_keypoints2;                                //第二次筛选结果
  81.     vector<KeyPoint> Final_Keypoints2;                                //第三次筛选结果
  83.     Mat descriptors_2;
  85.     double detect_2 = (double)getTickCount();
  87.     sift_1.detect(image_2, gauss_pyr_2, dog_pyr_2, keypoints_2, all_cell_contrasts_2,
  88.         average_contrast_2, n_cells_2, num_cell_2, available_n_2, available_num_2, final_keypoints2, Final_keypoints2, Final_Keypoints2);
  90.     cout << "待配准图像检测出的总特征数 =" << keypoints_2.size() << endl;
  92.     double detect_time_2 = ((double)getTickCount() - detect_2) / getTickFrequency();
  94.     cout << "待配准图像特征点检测时间是： " << detect_time_2 << "s" << endl;
  96.     double comput_2 = (double)getTickCount();
  98.     vector<Mat> sar_harris_fun_2;
  99.     vector<Mat> amplit_2;
  100.     vector<Mat> orient_2;
  102.     sift_1.comput_des(gauss_pyr_2, keypoints_2, amplit_2, orient_2, descriptors_2);
  104.     double comput_time_2 = ((double)getTickCount() - comput_2) / getTickFrequency();
  106.     cout << "待配准特征点描述时间是： " << comput_time_2 << "s" << endl;
  108.     /********************** 1、最近邻与次近邻距离比匹配，两幅影像进行配准 **********************/
  110.     myMatch mymatch;
  112.     double match_time = (double)getTickCount();                        //影像配准计时开始
  114.     //knnMatch函数是DescriptorMatcher类的成员函数，FlannBasedMatcher是DescriptorMatcher的子类
  115.     Ptr<DescriptorMatcher> matcher1 = new FlannBasedMatcher;
  116.     Ptr<DescriptorMatcher> matcher2 = new FlannBasedMatcher;
  118.     vector<vector<DMatch>> dmatchs;                                    //vector<DMatch>中存放的是一个描述子可能匹配的候选集
  119.     vector<vector<DMatch>> dmatch1;
  120.     vector<vector<DMatch>> dmatch2;
  122.     matcher1->knnMatch(descriptors_1, descriptors_2, dmatchs, 2);
  124.     cout << "距离比之前初始匹配点对个数是：" << dmatchs.size() << endl;
  126.     Mat matched_lines;                                                //同名点连线
  127.     vector<DMatch> init_matchs, right_matchs;                        //用于存放正确匹配的点
  129.     //该函数返回的是空间映射模型参数
  130.     Mat homography = mymatch.match(image_1, image_2, dmatchs, keypoints_1, keypoints_2, change_model, right_matchs, matched_lines, init_matchs);
  131.     
  133.     double match_time_2 = ((double)getTickCount() - match_time) / getTickFrequency();
  135.     cout << "特征点匹配花费时间是： " << match_time_2 << "s" << endl;
  136.     cout << change_model << "变换矩阵是：" << endl;
  137.     cout << homography << endl;
  139.     /********************** 1、把正确匹配点坐标写入文件中 **********************/
  141.     ofstream ofile;
  142.     ofile.open(".\\position.txt");                                    //创建文件
  143.     for (size_t i = 0; i < right_matchs.size(); ++i)
  144.     {
  145.         ofile << keypoints_1[right_matchs[i].queryIdx].pt << "   "
  146.             << keypoints_2[right_matchs[i].trainIdx].pt << endl;
  147.     }
  149.     /********************** 1、图像融合 **********************/
  151.     double fusion_beg = (double)getTickCount();
  153.     Mat fusion_image, mosaic_image, regist_image;
  154.     mymatch.image_fusion(image_1, image_2, homography, fusion_image, regist_image);
  156.     imwrite(".\\image_save\\融合后的图像.jpg", fusion_image);
  158.     double fusion_time = ((double)getTickCount() - fusion_beg) / getTickFrequency();
  159.     cout << "图像融合花费时间是： " << fusion_time << "s" << endl;
  161.     double total_time = ((double)getTickCount() - total_count_beg) / getTickFrequency();
  162.     cout << "总花费时间是： " << total_time << "s" << endl;
  164.     /********************** 1、生成棋盘图，显示配准结果 **********************/
  166.     //生成棋盘网格图像
  167.     Mat chessboard_1, chessboard_2, mosaic_images;
  169.     Mat image1 = imread(".\\image_save\\配准后的参考图像.jpg", -1);
  170.     Mat image2 = imread(".\\image_save\\配准后的待配准图像.jpg", -1);
  172.     mymatch.mosaic_map(image1, image2, chessboard_1, chessboard_2, mosaic_images, 50);
  174.     imwrite(".\\image_save\\参考图像棋盘图像.jpg", chessboard_1);
  175.     imwrite(".\\image_save\\待配准图像棋盘图像.jpg", chessboard_2);
  176.     imwrite(".\\image_save\\两幅棋盘图像叠加.jpg", mosaic_images);
  178.     return 0;
  179. }

四、SIFT 算法实验结果

1、实验数据

**说明：该数据是从网上下载的两个时相影像，左侧为旧时相影像，右侧为新时相影像； **

2、实验结果

(1)特征点检测结果

（2）最终匹配结果

（3）配准效果

说明：把两张图像交叉重叠在一起，可以观察其配准效果；

** 五、配置说明**

** 该结果是在 VS2019 + opencv4.4.0 + Debug x64 环境下跑出来的；**

** **如果感觉复制麻烦，完整工程见：SIFT算法C++代码实现-C/C++文档类资源-CSDN下载

标签： 计算机视觉 算法 人工智能
本文转载自: https://blog.csdn.net/weixin_47156401/article/details/122367593
版权归原作者 一米九零小胖子 所有， 如有侵权，请联系我们删除。
登录后发布评论

“SIFT算法详解(附有完整代码)”的评论:

还没有评论
关于作者
...
overfit同步小助手
文章同步
相关阅读

网络安全法-网络运行安全

使用selenium/drissionpage时如何阻止chrome自动跳转http到https

docker desktop 里部署的Open WebUI 管理员密码忘记了的处理方法

在ubuntu20.04中搭建onsite比赛运行环境

利用开源的低代码表单设计器FcDesigner高效管理和渲染复杂表单结构

Kafka学习笔记

【前端】浏览器输入url到页面呈现发生了什么？