昨天创建了新的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

posted on 2017-02-08 20:21  2师兄不会胖  阅读(1769)  评论(0编辑  收藏  举报