compute_sift_simple.cxx File Reference

#include "tests_config.hpp"
#include <config.hpp>
#include <utils/filesystem.hpp>
#include <utils/dataset.hpp>
#include <utils/vision.hpp>
#include <utils/logger.hpp>
#include <iostream>

Include dependency graph for compute_sift_simple.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 20 of file compute_sift_simple.cxx.

References vision::compute_sparse_sift_feature(), filesystem::create_file_directory(), filesystem::file_exists(), SimpleDataset::SimpleImage::location(), s_oxfordmini_data_dir, s_oxfordmini_database_location, and filesystem::write_cvmat().

                                 {
        LINFO << "HJ";
#if ENABLE_MULTITHREADING && ENABLE_MPI
        MPI::Init(argc, argv);
        int rank = MPI::COMM_WORLD.Get_rank();
        if(rank == 0) {
#endif
        SimpleDataset simple_dataset(s_oxfordmini_data_dir, s_oxfordmini_database_location);
        LINFO << simple_dataset;
#if ENABLE_MULTITHREADING && ENABLE_OPENMP
#pragma omp parallel for schedule(dynamic)
#endif
        for (int64_t i = 0; i < simple_dataset.num_images(); i++) {

                std::shared_ptr<SimpleDataset::SimpleImage> image = std::static_pointer_cast<SimpleDataset::SimpleImage>(simple_dataset.image(i));
                if (image == nullptr) continue;

                const std::string &keypoints_location = simple_dataset.location(image->feature_path("keypoints"));
                const std::string &descriptors_location = simple_dataset.location(image->feature_path("descriptors"));
                // if (filesystem::file_exists(keypoints_location) && filesystem::file_exists(descriptors_location)) continue;
                
                const std::string &image_location = simple_dataset.location(image->location());

                if (!filesystem::file_exists(image_location)) continue;
                
                cv::Mat im = cv::imread(image_location, cv::IMREAD_GRAYSCALE);

                cv::Mat keypoints, descriptors;
                if (!vision::compute_sparse_sift_feature(im, nullptr, keypoints, descriptors)) continue;

                filesystem::create_file_directory(keypoints_location);
                filesystem::create_file_directory(descriptors_location);

                filesystem::write_cvmat(keypoints_location, keypoints);
                filesystem::write_cvmat(descriptors_location, descriptors);
        }
#if ENABLE_MULTITHREADING && ENABLE_MPI
        }
        MPI::Finalize();
#endif
        return 0;
}