文件夹下的文件名路径读取（C++）

文本路径读取

在机器学习模型训练前期，需要对数据、图像、文本等进行预处理，而如何快速生成数据的文本路径呢？本文接下来直接使用C++实现文本路径生成，可查找固定格式如.jpg、.txt 等文件路径（绝对路径或文件名），然后保存为.txt 文本，方便后期图片数据读取使用。

---

C++代码实现如下：

#include <io.h>
#include <fstreanm>
#include <string>
#include <vector>
#include <iostream>
using namespace std;

void GetAllFiles(std::string path, std::vector<std::string>& files, std::string format)
{
	long long hfile = 0;
	struct _finddata_t fileinfo;
	std::string p;
	if ((hfile = _findfirst(p.assign(path).append("\\*" + format).c_str(), &fileinfo)) != -1)
	{
		do
		{
			//files.push_back(p.assign(fileinfo.name)); //save file name
			files.push_back(p.assign(path).append("\\").append(fileinfo.name)); // save file path
		} while (_findnext(hfile, &fileinfo) == 0);
		_findclose(hfile);
	}
	else if ((hfile = _findfirst(p.assign(path).append("\\*").c_str(), &fileinfo)) != -1)
	{
		do 
		{
			if ((fileinfo.attrib & _A_SUBDIR))
			{
				if (strcmp(fileinfo.name, ".") != 0 && strcmp(fileinfo.name, "..") != 0)
				{
					GetAllFiles(p.assign(path).append("\\").append(fileinfo.name), files, format);
				}
			}
			else
			{
				//files.push_back(p.assign(fileinfo.name));  
				//files.push_back(p.assign(path).append("\\").append(fileinfo.name));
			}
		} while (_findnext(hfile, &fileinfo) == 0);
		_findclose(hfile);
	}
}

int main()
{
	string filepath = "D:\\path..."; //文件根目录
	vector<string> files;
	char *dstAll = "path.txt";
	//读取所以格式为jpg的文件
	string format = ".jpg";
	GetAllFiles(filepath, files, format);
	ofstream ofn(distAll);
	int size = files.size();
	for(int i = 0; i<size; i++)
	{
		ofn<<files[i]<<endl; //写入文件
		cout<<files[i]<<endl; //输出到屏幕
	}
	ofn.close();
	cout<<"file number: "<<size<<endl;
	system("pause";)
	return 0;
}

---

注意：如果format赋值出错会进入死循环。