字符特征训练

typedef cv::Mat (*FEATURE_EXTRACTOR)(cv::Mat);
// Label 生成函数: 输入子文件夹名称,输出label
typedef double (*LABEL_MAKER)(std::string);
// 生成训练文件
void make_train_file(std::string src_dir, std::string dst,
	FEATURE_EXTRACTOR fea_extractor, LABEL_MAKER label_maker,
	int maxnum_per_class, bool shuffle)
{
	std::ofstream fs(dst.c_str());
	std::vector<std::string> sub_dirs;
	// 读入多个待训练图像源
	src_dir = TrimPath(src_dir);
	FindSubDirs(src_dir, sub_dirs);
	for (size_t i = 0; i != sub_dirs.size(); ++i) 
	{
		std::string sub = src_dir + "/" + sub_dirs[i];
		std::vector<std::string> img_paths;
		FindImages(sub, img_paths);
		// 随机选取
		if (shuffle) 
		{
			std::random_shuffle(img_paths.begin(), 
				img_paths.end());
		}
		if (img_paths.size() > maxnum_per_class) 
		{
			img_paths.erase(img_paths.begin() + 
				maxnum_per_class, img_paths.end());
		}
		for (size_t j = 0; j != img_paths.size(); ++j) 
		{
			// 训练文件每行的结构 
			// 实数的类标签——维数:该维数据
			fs << label_maker(sub_dirs[i]) << " ";
			// std::cout << sub_dirs[i] << "\n";
			cv::Mat pic = cv::imread(img_paths[j]);
			// 特征提取
			cv::Mat fea = fea_extractor(pic);
			float* dptr = (float*)fea.data;
			for (size_t k = 0; k != fea.total(); ++k, ++dptr) 
			{
				fs << (k + 1) << ":" << *dptr << " ";
			}
			fs << "\n";
		}
	}
	fs.close();
}




转载:http://blog.youkuaiyun.com/zhuwei1988

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值