运行环境:vs2013+opencv2.4.9+win10
数据来源于GTSRB。
模型训练的C++代码
#include<opencv2/opencv.hpp>
#include <iostream>
#include <fstream>
#include <vector>
#include <string>
#include<time.h>
using namespace cv;
using namespace std;
int main()
{
int imgWidht = 48;//重新定义图片大小48*48
int imgHeight = 48;
vector<string> imgTrainPath;//输入文件名变量
vector<int> imgTrainLabel;
int nLine = 0;
string buf;
ifstream imagePath("E:\\vs2013\\opencv_code\\GTSRBtrafficSign\\train\\trainImgPath.txt");//训练数据位置
unsigned long n;
while (imagePath)//读取训练样本,imageName.txt一行为路径,一行为标签,循环
{
if (getline(imagePath, buf))
{
nLine++;
if (nLine % 2 == 0)
{
imgTrainLabel.push_back(atoi(buf.c_str()));//atoi将字符串转换成整型,图片标签
}
else
{
imgTrainPath.push_back(buf);//图像路径
}
}
}