昨天创建了新的c++工程调用caffe,用训练好的caffemodel对图片进行分类。后来想批量的处理图片,因此去学习了下c++如何批量获取指定路径下的图片的完成路径。代码如下:
#include <io.h> #include <fstream> #include <string> #include <vector> #include <iostream> using namespace std;
//path为指定的路径,files是一个vector向量,里面存储所有图片的完整路径 void getAllFiles(string path, vector<string>& files) { //文件句柄 long hFile = 0; //文件信息 struct _finddata_t fileinfo; //较少用的文件信息读取结构 string p; 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) { files.push_back(p.assign(path).append("\\").append(fileinfo.name)); getAllFiles(p.assign(path).append("\\").append(fileinfo.name), files); } } else { files.push_back(p.assign(path).append("\\").append(fileinfo.name)); } } while (_findnext(hFile,&fileinfo)==0); //寻找下一个,成功返回0,否则-1 _findclose(hFile); } }
调用上面的getAllFiles(string path, vector<string>& files)函数即可获得完整路径,下面是我自己的一个示例:
void main() { char* filepath = "C:\\Users\\Administrator\\Desktop\\Img"; vector<string> files; getAllFiles(filepath, files); int size = files.size(); for (int i = 0; i < size; i++) { cout << files[i] << endl; } system("pause"); }
运行后的结果为:
每个图片的完整路径都已获得,但是图片的顺序不对这个原因我还没具体去研究。等研究好了,再补充。
PS:如果想了解_finddata_t的具体用法,请参考http://blog.csdn.net/xuejiren/article/details/37040827