sift/sift.cpp at master · thisaway/sift · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
#include "sift.h"
using namespace cv;

sift::sift(Mat img, int intervals)
{
	src_img = img.clone();
	num_intervals = intervals;
	num_octaves = floor(log(min(src_img.cols*2, src_img.rows*2)) / log(2) - 2);

	GenerateLists();
}

sift::sift(const char* filename, int intervals)
{
	src_img = imread(filename);

	num_intervals = intervals;
	num_octaves = floor(log(min(src_img.cols*2, src_img.rows*2)) / log(2) - 2);

	GenerateLists();
}

void sift::GenerateLists()
{
	unsigned int i = 0;

	g_list.resize(num_octaves);
	for (i = 0; i<num_octaves; i++)
		g_list[i].resize(num_intervals + 3);

	// 储存差分高斯金字塔的数列
	dog_list.resize(num_octaves);
	for (i = 0; i<num_octaves; i++)
		dog_list[i].resize(num_intervals + 2);

	// 生成数列去储存模糊系数
	abs_sigma = new double[num_intervals + 3];
	magnitude.resize(num_octaves);
	orientation.resize(num_octaves);
}

void sift::dosift()
{
	BuildScaleSpace();
	DetectExtrema();
	AssignOrientations();
	GenerateDescriptors();
}