Homec4science
Diffusion iCAPs public version 2 (master) Browse

iCAPs_v2/packages/gspbox/3rdparty/sources/flann-1.8.4-src/test/flann_kmeans_test.cpp
e5f5005bb2efmaster

Last Change
Raw File

flann_kmeans_test.cpp
View Options

#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 KMeans_SIFT10K : public DatasetTestFixture<float, float> {
protected:
KMeans_SIFT10K() : DatasetTestFixture("sift10K.h5") {}
};
TEST_F(KMeans_SIFT10K, TestSearch)
{
TestSearch<flann::L2<float> >(data, flann::KMeansIndexParams(7, 3, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(128), 0.75, gt_indices);
}
TEST_F(KMeans_SIFT10K, TestAddIncremental)
{
TestAddIncremental<flann::L2<float> >(data, flann::KMeansIndexParams(7, 3, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(110), 0.75, gt_indices);
}
TEST_F(KMeans_SIFT10K, TestAddIncremental2)
{
TestAddIncremental2<flann::L2<float> >(data, flann::KMeansIndexParams(7, 3, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(110), 0.75, gt_indices);
}
TEST_F(KMeans_SIFT10K, TestRemove)
{
TestRemove<flann::L2<float> >(data, flann::KMeansIndexParams(7, 3, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(128));
}
TEST_F(KMeans_SIFT10K, TestSave)
{
TestSave<flann::L2<float> >(data, flann::KMeansIndexParams(7, 3, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(128), 0.75, gt_indices);
}
TEST_F(KMeans_SIFT10K, TestCopy)
{
TestCopy<flann::L2<float> >(data, flann::KMeansIndexParams(7, 3, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(128), 0.75, gt_indices);
}
TEST_F(KMeans_SIFT10K, TestCopy2)
{
TestCopy2<KMeansIndex<flann::L2<float> > >(data, flann::KMeansIndexParams(7, 3, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(128), 0.75, gt_indices);
}
/**
* Test fixture for SIFT 100K dataset
*/
class KMeans_SIFT100K : public DatasetTestFixture<float, float> {
protected:
KMeans_SIFT100K() : DatasetTestFixture("sift100K.h5") {}
};
TEST_F(KMeans_SIFT100K, TestSearch)
{
TestSearch<flann::L2<float> >(data, flann::KMeansIndexParams(32, 11, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(96), 0.75, gt_indices);
}
TEST_F(KMeans_SIFT100K, TestAddIncremental)
{
TestAddIncremental<flann::L2<float> >(data, flann::KMeansIndexParams(32, 11, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(128), 0.75, gt_indices);
}
TEST_F(KMeans_SIFT100K, TestAddIncremental2)
{
TestAddIncremental2<flann::L2<float> >(data, flann::KMeansIndexParams(32, 11, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(128), 0.75, gt_indices);
}
TEST_F(KMeans_SIFT100K, TestRemove)
{
TestRemove<flann::L2<float> >(data, flann::KMeansIndexParams(32, 11, FLANN_CENTERS_RANDOM, 0.4), query, indices,
dists, knn, flann::SearchParams(128) );
}
TEST_F(KMeans_SIFT100K, TestSave)
{
TestSave<flann::L2<float> >(data, flann::KMeansIndexParams(32, 11, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(96), 0.75, gt_indices);
}
/**
* Test fixture for SIFT 10K dataset with byte feature elements
*/
class KMeans_SIFT10K_byte : public DatasetTestFixture<unsigned char, float> {
protected:
KMeans_SIFT10K_byte() : DatasetTestFixture("sift10K_byte.h5") {}
};
TEST_F(KMeans_SIFT10K_byte, TestSearch)
{
TestSearch<flann::L2<unsigned char> >(data, flann::KMeansIndexParams(7, 3, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(128), 0.75, gt_indices);
}
class KMeans_SIFT100K_byte : public DatasetTestFixture<unsigned char, float> {
protected:
KMeans_SIFT100K_byte() : DatasetTestFixture("sift100K_byte.h5") {}
};
TEST_F(KMeans_SIFT100K_byte, TestSearch)
{
TestSearch<flann::L2<unsigned char> >(data, flann::KMeansIndexParams(32, 11, FLANN_CENTERS_RANDOM, 0.4),
query, indices, dists, knn, flann::SearchParams(80), 0.75, gt_indices);
}
int main(int argc, char** argv)
{
testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}
c4science · Help