void build_model() {
vector<Mat> trainDesc;
FeatureDetector *detector = new SurfFeatureDetector();
DescriptorExtractor *extractor = new SurfDescriptorExtractor();
// Generate descriptors from the image db
fs::path p = fs::system_complete(IMAGE_DATABASE);
fs::directory_iterator end_iter;
for(fs::directory_iterator dir_itr(p); dir_itr != end_iter; ++dir_itr) {
string img_name(dir_itr->path().string());
Mat img = imread(img_name, CV_LOAD_IMAGE_GRAYSCALE);
// feature extraction
vector<KeyPoint> keypoints;
detector->detect(img, keypoints);
// feature description
Mat descriptor;
extractor->compute(img, keypoints, descriptor);
trainDesc.push_back(descriptor);
imgNames.push_back(img_name);
}
// train the model
matcher->add(trainDesc);
matcher->train();
// Clean up
delete detector;
delete extractor;
}
