OpenCV编程实例之图像文件批量读取。
本博文摘录《OpenCV图像处理编程实例》2.4章节，更详细的内容请参考本书。
在进行图片序列处理时，我们常常需要读取文件夹下的每一个图片，然后再进行分析处理，因此需要对文件名连续及无规则情况分开讨论。对于文件名连续的情况，文件读取就简单得多，可以利用sprintf函数实现在窗口中连续读取同一文件夹下的图片序列，而对于无规则的情况则可以采用基于C++下WIN32_ FIND_DATA文件的读取方式。
2.4.6 图像批量读取——规则
文件名连续情况下的读取如代码2-32所示。
// 功能：代码 2-32 文件名连续情况下 // 作者：朱伟 zhu1988wei@163.com // 来源：《OpenCV图像处理编程实例》 // 博客：http://blog.csdn.net/zhuwei1988 // 更新：2016-8-1 // 说明：版权所有，引用或摘录请联系作者，并按照上面格式注明出处，谢谢。// #include <iostream> #include <stdio.h> #include <stdlib.h> #include <opencv2/highgui/highgui.hpp> #include <opencv2/imgproc/imgproc.hpp> using namespace cv; using namespace std; int main() { // 定义相关参数 const int num = 4; char fileName[50]; char windowName[50]; cv::Mat srcImage; for (int i = 1; i <= num; i++) { // sprintf读入指定路径下图片序列 sprintf_s(fileName, "..\\images\\test\\1 (%d).jpg", i); sprintf_s(windowName, "NO%d", i); // 按照图像文件名读取 srcImage = cv::imread(fileName); if (!srcImage.data) { std::cout << "No data!" << std::endl; return -1; } cv::namedWindow(windowName); cv::imshow(windowName, srcImage); std::cout << "NO: " << i << std::endl; //cv::waitKey(0); /* 该处可以添加处理步骤 */ } cv::waitKey(0); return 0; }
第16行代码利用sprintf将对应的图像文件路径转换为char*，在这种文件名连续的时候可以选中文件夹中的所有图像文件，然后用鼠标右键选定并重命名，键入1后，文件夹的所有文件自动命名为1 (k).jpg，其中k取值为1，2……然后就可以根据本方法进行批量读取了。
2.4.7 图像批量读取——无规则
文件名无规则的情况读取如代码2-33所示。
// 功能：代码 2-33 文件名无规则情况读取 // 作者：朱伟 zhu1988wei@163.com // 来源：《OpenCV图像处理编程实例》 // 博客：http://blog.csdn.net/zhuwei1988 // 更新：2016-8-1 // 说明：版权所有，引用或摘录请联系作者，并按照上面格式注明出处，谢谢。// #include <opencv2/core/core.hpp> #include <opencv2/highgui/highgui.hpp> #include <opencv2/imgproc/imgproc.hpp> #include <iostream> #include <stdio.h> #include <windows.h> using namespace std; // LPCWSTR转string std::string WChar2Ansi(LPCWSTR pwszSrc) { int nLen = WideCharToMultiByte(CP_ACP, 0, pwszSrc, -1, NULL, 0, NULL, NULL); if (nLen <= 0) return std::string(""); char* pszDst = new char[nLen]; if (NULL == pszDst) return std::string(""); WideCharToMultiByte(CP_ACP, 0, pwszSrc, -1, pszDst, nLen, NULL, NULL); pszDst[nLen - 1] = 0; std::string strTemp(pszDst); delete[] pszDst; return strTemp; } // 利用winWIN32_FIND_DATA读取文件下的文件名 void readImgNamefromFile(char* fileName, vector <string> &imgNames) { // vector清零 参数设置 imgNames.clear(); WIN32_FIND_DATA file; int i = 0; char tempFilePath[MAX_PATH + 1]; char tempFileName[50]; // 转换输入文件名 sprintf_s(tempFilePath, "%s/*", fileName); // 多字节转换 WCHAR wstr[MAX_PATH] = { 0 }; MultiByteToWideChar(CP_ACP, 0, tempFilePath, -1, wstr, sizeof(wstr)); // 查找该文件待操作文件的相关属性读取到WIN32_FIND_DATA HANDLE handle = FindFirstFile(wstr, &file); if (handle != INVALID_HANDLE_VALUE) { FindNextFile(handle, &file); FindNextFile(handle, &file); // 循环遍历得到文件夹的所有文件名 do { sprintf(tempFileName, "%s", fileName); imgNames.push_back(WChar2Ansi(file.cFileName)); imgNames[i].insert(0, tempFileName); i++; } while (FindNextFile(handle, &file)); } FindClose(handle); } int main() { // 设置读入图像序列文件夹的路径 char* fileName = "..\\images\\test\\"; std::vector <string> imgNames; // 获取对应文件夹下所有文件名 readImgNamefromFile(fileName, imgNames); // 遍历对应文件夹下所有文件名 for (int i = 0; i < imgNames.size(); i++) { cv::Mat img = cv::imread(imgNames[i]); if (!img.data) return -1; /* 可添加图像处理算法code*/ cv::imshow("im", img); cv::waitKey(0); } return 0; }
利用winWIN32_FIND_DATA读取文件夹下文件的思路：首先转换文件夹名，利用FindFirstFile获取当前文件夹名的句柄；然后遍历当前文件夹名下的所有文件，将得到的所有文件名称转换后赋值于图像文件向量；最后遍历完当前文件下的所有文件，生成相应图像文件索引名称，用于文件夹中所有图像文件的读取，在读取单个图像文件后可进行相关的图像处理操作。
每天告诉自己我很棒!
