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