BAFT/write_descriptors.cpp at master · arnfred/BAFT · GitHub

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
/**
 * @file write_descriptors.cpp
 * @brief Write out descriptors for baft
 * @date December 7, 2014
 * @author Jonas Toft Arnfred
 */

#include "./src/utils.h"
#include "./src/baft.h"

// System
#include <string>
#include <vector>
#include <iostream>
#include <fstream>

using namespace std;
using namespace cv;

/* ************************************************************************* */
int main(int argc, char *argv[]) {

  cv::Mat imgN;

  // Define sizes and other variables
  std::vector<int> sizes { 16, 32, 64, 128 };
  //std::vector<int> sizes { 1024 };
  std::vector<string> testsets { "bark", "bikes", "boat", "graf", "leuven", "trees", "ubc", "wall", "abs_x1", "abs_x4", "abs_x10", "trans_t2", "trans_t4" };

  int limit = 3000; // default value
  int blur = 0; // default value
  bool rotate = 0; // default value
  int patchSize = 30;
  if (argc > 1)
    limit = (int)atoi(argv[1]);
  if (argc > 2)
    blur = (int)atoi(argv[2]);
  if (argc > 3)
    rotate = (int)atoi(argv[3]);
  if (argc > 4)
    patchSize = (int)atoi(argv[4]);

  for (int j = 0; j < (int)sizes.size(); j++)
  {
    int size = sizes[j];

    for (int k = 0; k < (int)testsets.size(); k++)
    {

      // How many images?
      int nb_files = 6;
      const string testset = testsets[k];
      if (testset == "abs_x1" or testset == "abs_x4" or testset == "abs_x10" or testset == "trans_t2" or testset == "trans_t4")
        nb_files = 9;

      // Find file ending
      string fileEnding = ".ppm";
      if (testset == "boat" or testset == "abs_x1" or testset == "abs_x4" or testset == "abs_x10" or testset == "trans_t2" or testset == "trans_t4")
        fileEnding = ".pgm";

      for (int i = 0; i < nb_files; i++)
      {

        // Save descriptors from image `i`
        std::ostringstream imgNStream;
        imgNStream << "../data/" << testset << "/img" << (i+1) << fileEnding;
        const string imgNFile = imgNStream.str();
        imgN = imread(imgNFile, 1);

        cout << testset << "_" << (i+1) << ": " << size << " bytes\n";
        // Create baft object
        Ptr<Feature2D> dbaft;
        dbaft = BAFT::create(limit, size, patchSize, blur, rotate);
        // Detect baft features in the images
        vector<cv::KeyPoint> kptsN;
        cv::Mat descN;
        dbaft->detectAndCompute(imgN, cv::noArray(), kptsN, descN);

        // Write out descriptors
        std::ostringstream desc_file_name;
        desc_file_name << "../descriptors/" << testset << "_img" << (i+1) << "_" << size << ".dsc";
        ofstream desc_file;
        desc_file.open(desc_file_name.str());
        std::ostringstream pos_file_name;
        pos_file_name << "../descriptors/" << testset << "_img" << (i+1) << ".kp";
        ofstream pos_file;
        pos_file.open(pos_file_name.str());
        uchar* desc_p = descN.ptr<uchar>();
        for (int m = 0; m < descN.rows; m++)
        {
          for (int n = 0; n < descN.cols; n++)
          {
            desc_file << (int)desc_p[m*descN.cols + n] << " ";
          }
          desc_file << "\n";
          pos_file << kptsN[m].pt.x << " " << kptsN[m].pt.y << " "
                   << kptsN[m].response << " " << kptsN[m].size << " "
                   << kptsN[m].angle << " " << kptsN[m].octave << "\n";
        }
        desc_file.close();
        pos_file.close();
      }
    }
  }

  return 1;
}