【源.cpp代码】
#include
#include
using namespace cv;
using namespace std;
//定义一个boost库的命名空间
namespace fs = boost::filesystem;
using namespace fs;
class categorizer
{
private:
//从类目名称到数据的map映射
map<string, Mat> result_objects;
//存放所有训练图片的BOW
map<string, Mat> allsamples_bow;
//从类目名称到训练图集的映射,关键字可以重复出现
multimap<string, Mat> train_set;
// 训练得到的SVM
CvSVM *stor_svms;
//类目名称,也就是TRAIN_FOLDER设置的目录名
vector category_name;
//类目数目
int categories_size;
//用SURF特征构造视觉词库的聚类数目
int clusters;
//存放训练图片词典
Mat vocab;
//特征检测器detectors与描述子提取器extractors 泛型句柄类Ptr
Ptr<FeatureDetector> featureDecter;
Ptr<DescriptorExtractor> descriptorExtractor;
Ptr<BOWKMeansTrainer> bowtrainer;
Ptr<BOWImgDescriptorExtractor> bowDescriptorExtractor;
Ptr<FlannBasedMatcher> descriptorMacher;
//构造训练集合
void make_train_set();
// 移除扩展名,用来讲模板组织成类目
string remove_extention(string);
public:
//构造函数
categorizer(int);
// 聚类得出词典
void bulid_vacab();
//构造BOW
void compute_bow_image();
//训练分类器
void trainSvm();
//将测试图片分类
void category_By_svm();
};
// 移除扩展名,用来讲模板组织成类目
string categorizer::remove_extention(string full_name)
{
//find_last_of找出字符最后一次出现的地方
int last_index = full_name.find_last_of(".");
string name = full_name.substr(0, last_index);
return name;
}
// 构造函数
categorizer::categorizer(int _clusters)
{
cout << “开始初始化…” << endl;
clusters = _clusters;
//初始化指针
featureDecter = new SurfFeatureDetector();
descriptorExtractor = new SurfDescriptorExtractor();
bowtrainer = new BOWKMeansTrainer(clusters);
descriptorMacher = new FlannBasedMatcher();
bowDescriptorExtractor = new BOWImgDescriptorExtractor(descriptorEx