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);

// 循环遍历得到文件夹的所有文件名

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

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

相关文章

快网idc优惠网

相关文章

微信

快网idc优惠网

QQ交流群