680b1740b1
Added FindObject and ObjSignature classes Added ULogger from utilite for pretty log git-svn-id: http://find-object.googlecode.com/svn/trunk/find_object@356 620bd6b2-0a58-f614-fd9a-1bd335dccda9
36 lines
871 B
C++
36 lines
871 B
C++
/*
|
|
* Vocabulary.h
|
|
*
|
|
* Created on: 2014-05-09
|
|
* Author: mathieu
|
|
*/
|
|
|
|
#ifndef VOCABULARY_H_
|
|
#define VOCABULARY_H_
|
|
|
|
#include <QtCore/QMultiMap>
|
|
#include <QtCore/QVector>
|
|
#include <opencv2/opencv.hpp>
|
|
|
|
class Vocabulary {
|
|
public:
|
|
Vocabulary();
|
|
virtual ~Vocabulary();
|
|
|
|
void clear();
|
|
QMultiMap<int, int> addWords(const cv::Mat & descriptors, int objectId, bool incremental);
|
|
void update();
|
|
void search(const cv::Mat & descriptors, cv::Mat & results, cv::Mat & dists, int k);
|
|
int size() const {return indexedDescriptors_.rows + notIndexedDescriptors_.rows;}
|
|
const QMultiMap<int, int> & wordToObjects() const {return wordToObjects_;}
|
|
|
|
private:
|
|
cv::flann::Index flannIndex_;
|
|
cv::Mat indexedDescriptors_;
|
|
cv::Mat notIndexedDescriptors_;
|
|
QMultiMap<int, int> wordToObjects_; // <wordId, ObjectId>
|
|
QVector<int> notIndexedWordIds_;
|
|
};
|
|
|
|
#endif /* VOCABULARY_H_ */
|