OpenCV 应用读取文件路径与文件名批量处理图片

悠悠 2022-07-13 06:37 379阅读 0赞

在应用OpenCV大量测试图片时，需要对图片批量的读入并进行处理。之前处理这个问题时是使用这种方法：把待处理的图片放到一个文件夹内，全选它们然后重命名1，这样系统会自动给他们全部重命名为1(1)，1(2)，1(3)等等等  
然后用下面的代码把图片读进来：

for ( i=1;i<=624;i++)
    	{
    	sprintf_s(adr, "C:\Users\Administrator\Desktop\第二组截图\1 (%d).jpg",i);
    	Mat g_SrcImage;
        g_SrcImage=imread(adr);
    	printf("i=%d",i);
    	}

这种方法很麻烦，需要手动重命名一遍，然后根据文件夹下的图片个数确定循环中的值。有一种更简便并且灵活性更高的方法，就是遍历文件夹内所有图片的路径，名称和总个数。  
下面这种实现方式其实和OpenCV本身没什么关系了，是一种应用C++提供的`io.h`头文件中定义的函数实现。  
先给出函数的定义：

void listFiles(const char * dir, vector<string>& files);

可以看到函数没有返回值，而是将遍历到的文件信息存储到`vector<string>`中，完整的代码实现如下：

#include <iostream>
    #include <io.h>
    #include <vector>
    #include <string>
    #include <opencv2/opencv.hpp>
    #include <opencv2/core/core.hpp>
    #include <opencv2/highgui/highgui.hpp>
    
    using namespace std;
    using namespace cv;
    
    void listFiles(const char * dir, vector<string>& files);
    
    int main()
    {
    	string path = "E:\\facedata\\faceReg";
    	vector<string> files;
    	listFiles(path.c_str(),files);
    	for (int i = 0; i < files.size();i++)
    	{
    		cout << files[i] << endl;
    		Mat SrcImage = imread(files[i]);
    		namedWindow("show", 0);
    		imshow("show", SrcImage);
    		waitKey(10);
    	}
    	waitKey(0);
    	return 0;
    }
    //目录中的所有图片(到每一级目录)
    void listFiles(const char * dir, vector<string>& files)
    {
    	char dirNew[200];
    	strcpy(dirNew, dir);
    	strcat(dirNew, "\\*.*");    // 在目录后面加上"\\*.*"进行第一次搜索
    	intptr_t handle;
    	_finddata_t findData;
    	handle = _findfirst(dirNew, &findData);
    	if (handle == -1)        // 检查是否成功
    		return;
    	do
    	{
    		if (findData.attrib & _A_SUBDIR)
    		{
    			if (strcmp(findData.name, ".") == 0 || strcmp(findData.name, "..") == 0)
    				continue;
    			cout << findData.name << "\t<dir>\n";
    			// 在目录后面加上"\\"和搜索到的目录名进行下一次搜索
    			strcpy(dirNew, dir);
    			strcat(dirNew, "\\");
    			strcat(dirNew, findData.name);
    			listFiles(dirNew, files);
    		}
    		else
    			files.push_back(string(dir).append("\\").append(findData.name));
    			cout << findData.name << "\t" << findData.size << " bytes.\n";
    	} while (_findnext(handle, &findData) == 0);
    	_findclose(handle);    // 关闭搜索句柄
    }

运行结果：  
![这里写图片描述][70]

备注：  
1.在上面的代码中可以看到，`listFiles`函数其实在利用递归，这意味着，这个函数不仅仅可以找目录中的文件，还可以找到目录下每一层的文件，在大多数情况下并不需要区分是遍历目录下还是遍历目录中，因为目录是我们自己创建的，要遍历的路径也是自己输入，所以我们完全可以把这个当做遍历目录中文件的函数来用。  
2.上述代码在x64，x86平台上都测试通过，之所以出现x86平台运行正常，x64编译通过，运行出现异常，是因为\_findfirst()返回类型为intptr\_t而非long型，从“intptr\_t”转换到“long”丢失了数据，所以创建句柄时需要：`intptr_t handle;`

[70]: /images/20220713/7c3f3c5ff4e94d6ca21f86bce03a1607.png