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
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
#include <gtest/gtest.h>
#include <time.h>
#include <flann/flann.h>
#include <flann/io/hdf5.h>
#include "flann_tests.h"
using namespace flann;
/**
* Test fixture for SIFT 10K dataset
*/
class Linear_SIFT10K : public DatasetTestFixture<float, float> {
protected:
Linear_SIFT10K() : DatasetTestFixture("sift10K.h5") {}
};
TEST_F(Linear_SIFT10K, TestSearch)
{
TestSearch<flann::L2<float> >(data, flann::LinearIndexParams(),
query, indices, dists, knn, flann::SearchParams(0), 1.0, gt_indices);
}
TEST_F(Linear_SIFT10K, TestSearch2)
{
TestSearch2<flann::L2<float> >(data, flann::LinearIndexParams(),
query, indices, dists, knn, flann::SearchParams(0), 1.0, gt_indices);
}
TEST_F(Linear_SIFT10K, TestRemove)
{
TestRemove<flann::L2<float> >(data, flann::LinearIndexParams(),
query, indices, dists, knn, flann::SearchParams(0));
}
TEST_F(Linear_SIFT10K, TestSave)
{
TestSave<flann::L2<float> >(data, flann::LinearIndexParams(),
query, indices, dists, knn, flann::SearchParams(0), 1.0, gt_indices);
}
TEST_F(Linear_SIFT10K, TestCopy)
{
TestCopy<flann::L2<float> >(data, flann::LinearIndexParams(),
query, indices, dists, knn, flann::SearchParams(0), 1.0, gt_indices);
}
TEST_F(Linear_SIFT10K, TestCopy2)
{
TestCopy<flann::L2<float> >(data, flann::LinearIndexParams(),
query, indices, dists, knn, flann::SearchParams(0), 1.0, gt_indices);
}
/**
* Test fixture for SIFT 100K dataset
*/
class Linear_SIFT100K : public DatasetTestFixture<float, float> {
protected:
Linear_SIFT100K() : DatasetTestFixture("sift100K.h5") {}
};
TEST_F(Linear_SIFT100K, TestSearch)
{
TestSearch<flann::L2<float> >(data, flann::LinearIndexParams(),
query, indices, dists, knn, flann::SearchParams(0), 1.0, gt_indices);
}
/**
* Test fixture for SIFT 10K dataset with byte feature elements
*/
class Linear_SIFT10K_byte : public DatasetTestFixture<unsigned char, float> {
protected:
Linear_SIFT10K_byte() : DatasetTestFixture("sift10K_byte.h5") {}
};
TEST_F(Linear_SIFT10K_byte, Linear)
{
TestSearch<flann::L2<unsigned char> >(data, flann::LinearIndexParams(),
query, indices, dists, knn, flann::SearchParams(0), 1.0, gt_indices);
}
class Linear_SIFT100K_byte : public DatasetTestFixture<unsigned char, float> {
protected:
Linear_SIFT100K_byte() : DatasetTestFixture("sift100K_byte.h5") {}
};
TEST_F(Linear_SIFT100K_byte, TestSearch)
{
TestSearch<flann::L2<unsigned char> >(data, flann::LinearIndexParams(),
query, indices, dists, knn, flann::SearchParams(0), 1.0, gt_indices);
}
int main(int argc, char** argv)
{
testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}
|
