caffe训练时根据文件名产生类别标号文件c++代码

caffe训练时，图片分为test集和train集，图片如下：

要产生文件列表

格式如下:

test\300.jpg 3

test\301.jpg 3

test\400.jpg 4

test\401.jpg 4

test\500.jpg 5

test\501.jpg 5

代码如下： 

#include "stdafx.h"
#include <string>
#include <io.h>
#include <vector>
#include <iostream>

using namespace std;

/************************************************************************/
/*  获取文件夹下所有文件名
输入：
path	:	文件夹路径
exd		:   所要获取的文件名后缀，如jpg、png等；如果希望获取所有
文件名, exd = ""
输出：
files	:	获取的文件名列表

shao, 20140707
*/
/************************************************************************/
void getFiles(string path, string exd, vector<string>& files)
{
	//cout << "getFiles()" << path<< endl; 
	//文件句柄
	long   hFile = 0;
	//文件信息
	struct _finddata_t fileinfo;
	string pathName, exdName;

	if (0 != strcmp(exd.c_str(), ""))
	{
		exdName = "\\*." + exd;
	}
	else
	{
		exdName = "\\*";
	}

	if ((hFile = _findfirst(pathName.assign(path).append(exdName).c_str(), &fileinfo)) != -1)
	{
		do
		{
			//cout << fileinfo.name << endl; 

			//如果是文件夹中仍有文件夹,迭代之
			//如果不是,加入列表
			if ((fileinfo.attrib &  _A_SUBDIR))
			{
				if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
					getFiles(pathName.assign(path).append("\\").append(fileinfo.name), exd, files);
			}
			else
			{
				if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
					files.push_back(pathName.assign(path).append("\\").append(fileinfo.name));
			}
		} while (_findnext(hFile, &fileinfo) == 0);
		_findclose(hFile);
	}
}

void main()
{
	cout << "create train and test file list" << endl;
	vector<string> files;
	char * filePath = "D:\\deep_learning\\caffe-windows-master\\data\\re\\test";

	//获取该路径下的所有jpg文件
	getFiles(filePath, "jpg", files);

	FILE* fp;
	string test_file = "D:\\deep_learning\\caffe-windows-master\\examples\\myfile\\test.txt";
	errno_t tOk = fopen_s(&fp, test_file.c_str(), "w");
	if (0 != tOk){
		cout << "fopen_s error:" << tOk << endl;
		return; 
	}

	cout << test_file << endl; 

	int size = files.size();
	for (int i = 0; i < size; i++)
	{
		int pos = files[i].find_last_of("\\");
		string fname = files[i].substr(pos + 1);
		string f = fname.substr(0, 1);
		string row = "test\\" + fname + " " + f;
		if (i > 0){
			row = "\n" + row;
		}
		//cout << row << endl;
		fputs(row.c_str(), fp);

		//cout << "fx="<<files[i].c_str() << endl;
	}
	fclose(fp);
	 
	char * filePath2 = "D:\\deep_learning\\caffe-windows-master\\data\\re\\train";

	//获取该路径下的所有jpg文件
	//先要清空一下
	files.clear(); 
	getFiles(filePath2, "jpg", files);
	 
	string train_file = "D:\\deep_learning\\caffe-windows-master\\examples\\myfile\\train.txt";
	tOk = fopen_s(&fp, train_file.c_str(), "w");
	if (0 != tOk){
		cout << "train_file fopen_s error:" << tOk << endl;
		return;
	}
	cout << train_file << endl; 
	   
	size = files.size();
	cout << "train file size="<<size << endl;
	for (int i = 0; i < size; i++)
	{
		int pos = files[i].find_last_of("\\");
		string fname = files[i].substr(pos + 1);
		string f = fname.substr(0, 1);
		string row = "train\\" + fname + " " + f;
		if (i > 0){
			row = "\n" + row;
		}

		//cout << row << endl;
		fputs(row.c_str(), fp);

		//cout << "fx="<<files[i].c_str() << endl;
	}
	fclose(fp);


	getchar();
}