mirror/opencv
1
0
Fork 0
mirror of https://github.com/opencv/opencv.git synced 2025-06-12 04:12:52 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

added sample on BOW usage to image classification (training and testing is on Pascal VOC dataset)

Browse Source
This commit is contained in:
Maria Dimashova 2010-10-08 13:17:34 +00:00
parent 66df8ef06c
commit 9f934363e5
3 changed files with 2588 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
modules/features2d/include/opencv2/features2d/features2d.hpp
View File

@ -2243,6 +2243,9 @@ CV_EXPORTS void evaluateGenericDescriptorMatcher( const Mat& img1, const Mat& im
class CV_EXPORTS BOWTrainer class CV_EXPORTS BOWTrainer
{ {
public: public:
BOWTrainer(){}
virtual ~BOWTrainer(){}
void add( const Mat& descriptors ); void add( const Mat& descriptors );
const vector<Mat>& getDescriptors() const { return descriptors; } const vector<Mat>& getDescriptors() const { return descriptors; }
int descripotorsCount() const { return descriptors.empty() ? 0 : size; } int descripotorsCount() const { return descriptors.empty() ? 0 : size; }
@ -2272,8 +2275,7 @@ class CV_EXPORTS BOWKMeansTrainer : public BOWTrainer
public: public:
BOWKMeansTrainer( int clusterCount, const TermCriteria& termcrit=TermCriteria(), BOWKMeansTrainer( int clusterCount, const TermCriteria& termcrit=TermCriteria(),
int attempts=3, int flags=KMEANS_PP_CENTERS ); int attempts=3, int flags=KMEANS_PP_CENTERS );
virtual ~BOWKMeansTrainer(){}
// Returns trained vocabulary (i.e. cluster centers). // Returns trained vocabulary (i.e. cluster centers).
virtual Mat cluster() const; virtual Mat cluster() const;
@ -2295,6 +2297,8 @@ class CV_EXPORTS BOWImgDescriptorExtractor
public: public:
BOWImgDescriptorExtractor( const Ptr<DescriptorExtractor>& dextractor, BOWImgDescriptorExtractor( const Ptr<DescriptorExtractor>& dextractor,
const Ptr<DescriptorMatcher>& dmatcher ); const Ptr<DescriptorMatcher>& dmatcher );
virtual ~BOWImgDescriptorExtractor(){}
void setVocabulary( const Mat& vocabulary ); void setVocabulary( const Mat& vocabulary );
const Mat& getVocabulary() const { return vocabulary; } const Mat& getVocabulary() const { return vocabulary; }
void compute( const Mat& image, vector<KeyPoint>& keypoints, Mat& imgDescriptor, void compute( const Mat& image, vector<KeyPoint>& keypoints, Mat& imgDescriptor,

2
modules/features2d/src/bagofwords.cpp
View File

@ -84,7 +84,7 @@ Mat BOWKMeansTrainer::cluster() const
Mat mergedDescriptors( descCount, descriptors[0].cols, descriptors[0].type() ); Mat mergedDescriptors( descCount, descriptors[0].cols, descriptors[0].type() );
for( size_t i = 0, start = 0; i < descriptors.size(); i++ ) for( size_t i = 0, start = 0; i < descriptors.size(); i++ )
{ {
Mat submut = mergedDescriptors.rowRange(start, descriptors[i].rows); Mat submut = mergedDescriptors.rowRange(start, start + descriptors[i].rows);
descriptors[i].copyTo(submut); descriptors[i].copyTo(submut);
start += descriptors[i].rows; start += descriptors[i].rows;
} }

2581
samples/cpp/bagofwords_classification.cpp Normal file
View File

File diff suppressed because it is too large Load Diff