Permalink
Switch branches/tags
Nothing to show
Find file Copy path
Fetching contributors…
Cannot retrieve contributors at this time
49 lines (40 sloc) 1.25 KB
#include <iostream>
#include <fstream>
#include <vector>
#include <boost/program_options.hpp>
#include <boost/assert.hpp>
#include <kgraph.h>
#include <kgraph-data.h>
using namespace std;
int main (int argc, char *argv[]) {
string input_path;
string output_path;
namespace po = boost::program_options;
po::options_description desc("Allowed options");
desc.add_options()
("help,h", "produce help message.")
("input,I", po::value(&input_path), "")
("output,O", po::value(&output_path), "")
;
po::positional_options_description p;
p.add("input", 1);
p.add("output", 1);
po::variables_map vm;
po::store(po::command_line_parser(argc, argv).options(desc).positional(p).run(), vm);
po::notify(vm);
if (vm.count("help") || (vm.count("input") < 1) || (vm.count("output") < 1))
{
cout << "lshkit2fvec <input> <output>" << endl;
return 0;
}
kgraph::Matrix<float> data;
data.load_lshkit(input_path);
ofstream os(output_path.c_str(), ios::binary);
int d = data.dim();
for (unsigned i = 0; i < data.size(); ++i) {
os.write((char const *)&d, sizeof(d));
os.write((char const *)data[i], sizeof(float) * d);
}
os.close();
return 0;
}