OpenCV图像文件批量读取编程实例

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获取当前文件夹名的句柄；然后遍历当前文件夹名下的所有文件，将得到的所有文件名称转换后赋值于图像文件向量；最后遍历完当前文件下的所有文件，生成相应图像文件索引名称，用于文件夹中所有图像文件的读取，在读取单个图像文件后可进行相关的图像处理操作。