如何快速生成数据的文本路径呢?C++实现文本路径生成
时间:2018-08-06 15:10
在机器学习模型训练前期,需要对数据、图像、文本等进行预处理,而如何快速生成数据的文本路径呢?本文接下来直接使用C++实现文本路径生成,可查找固定格式如 注意:如果format赋值出错会进入死循环。 相关文章: 如何实现浏览获取本地文件路径 C#如何使用浏览按钮获得文件路径和文件夹路径的实现方法 以上就是如何快速生成数据的文本路径呢?C++实现文本路径生成的详细内容,更多请关注gxlsystem.com其它相关文章!文本路径读取
.jpg
、.txt
等文件路径(绝对路径或文件名),然后保存为.txt
文本,方便后期图片数据读取使用。C++代码实现如下:
#include <io.h>#include <fstreanm>#include <string>#include <vector>#include <iostream>using namespace std;void GetAllFiles(string path, vector<string>& files, string format)
{ long hfile = 0; struct _finddata_t fileinfo; //用来存储文件信息的结构体
string p; if(hfile = _findfirst(p.assign(path).append("\\*" + format).c_str(), &fileinfo)) != -1) //第一次查找
{ do{ //files.push_back(p.assign(fileinfo.name)); //只保存文件名files.push_back(p.assign(path).appand("\\").append(fileinfo.name)); //保存文件路径和文件名
}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).appand("\\").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;
}