bench_index.cxx File Reference

#include "bench_config.hpp"
#include <utils/filesystem.hpp>
#include <utils/numerics.hpp>
#include <utils/dataset.hpp>
#include <utils/vision.hpp>
#include <utils/logger.hpp>
#include <search/bag_of_words/bag_of_words.hpp>
#include <search/inverted_index/inverted_index.hpp>
#include <iostream>

Include dependency graph for bench_index.cxx:

Go to the source code of this file.

Functions
_INITIALIZE_EASYLOGGINGPP int	main (int argc, char *argv[])

Function Documentation

_INITIALIZE_EASYLOGGINGPP int main	(	int	argc,
		char *	argv[]
	)

Definition at line 22 of file bench_index.cxx.

References filesystem::create_file_directory(), Dataset::location(), s_oxford_data_dir, s_oxford_database_location, s_oxford_num_clusters, and InvertedIndex::train().

                                 {
#if ENABLE_MULTITHREADING && ENABLE_MPI
        MPI::Init(argc, argv);
#endif
        const uint32_t num_clusters = s_oxford_num_clusters;

        SimpleDataset oxford_dataset(s_oxford_data_dir, s_oxford_database_location);
        LINFO << oxford_dataset;

        std::stringstream vocab_output_file;
        vocab_output_file << oxford_dataset.location() << "/vocabulary/" << num_clusters << ".vocab";

        std::shared_ptr<BagOfWords> bow = std::make_shared<BagOfWords>(vocab_output_file.str());

        InvertedIndex ii;
        std::shared_ptr<InvertedIndex::TrainParams> train_params = std::make_shared<InvertedIndex::TrainParams>();
        train_params->bag_of_words = bow;
        ii.train(oxford_dataset, train_params, oxford_dataset.all_images());

        std::stringstream index_output_file;
        index_output_file << oxford_dataset.location() << "/index/" << num_clusters << ".index";
        filesystem::create_file_directory(index_output_file.str());
        ii.save(index_output_file.str());

#if ENABLE_MULTITHREADING && ENABLE_MPI
        MPI::Finalize();
#endif
        return 0;
}