generic_tree.h

Go to the documentation of this file.

#ifndef VOCABULARY_TREE_GENERIC_TREE_H
#define VOCABULARY_TREE_GENERIC_TREE_H

#include "vocabulary_tree/vocabulary_tree.h"
#include <opencv/cv.h>

namespace vt {

namespace distance {
template<> struct L2<cv::Mat>
{
  typedef double result_type;

  result_type operator()(const cv::Mat& a, const cv::Mat& b) const
  {
    return cv::norm(a, b);
  }
};

} //namespace distance

class GenericTree : public VocabularyTree<cv::Mat>
{
public:
  GenericTree();
  GenericTree(const std::string& file);

  void save(const std::string& file) const;
  void load(const std::string& file);

  size_t dimension() const;
};

} //namespace vt

#endif