xp/search_database.h at master · brly/xp · 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
#ifndef SEARCH_DATABASE_H_
#define SEARCH_DATABASE_H_

// "Search Database" という名前のクラスだが、
// 検索の際には、クエリとなる、"クエリの特徴ベクトルの重み" wqが必要となるため
// 名前の割には汎用的ではない事に注意する
// そのため、クラス名の変更が考えられるが、良い名前を思いつかなかった

#include <vector>
#include <string>
#include <utility>

class SearchDatabase {
  typedef std::pair<double, std::string> ResultPair;
  typedef std::vector<double> Vec;

  static std::vector<ResultPair> result;
  static std::vector<Vec> database_vector;
  static std::vector<std::string> database_string;

  static void calc_in_database(const std::vector<double>& wq, const int top_k);
  static void show_images(const std::vector<std::string>& files);

 public:
  // 検索を行い、結果を標準出力へ出力
  static void search(const std::vector<double>& wq, const int top_k = 20);
  // 検索を行い、結果を引数へ与え、出力は行わない
  static void search(const std::vector<double>& wq, std::vector<std::string>& dest,
                     const int top_k = 20);

  static void init();
};

#endif // SEARCH_DATABASE_H_