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获取当前文件夹名的句柄;然后遍历当前文件夹名下的所有文件,将得到的所有文件名称转换后赋值于图像文件向量;最后遍历完当前文件下的所有文件,生成相应图像文件索引名称,用于文件夹中所有图像文件的读取,在读取单个图像文件后可进行相关的图像处理操作。
您可能感兴趣的文章:
- OpenCV中的cv::Mat函数将数据写入txt文件
- OpenCV cv.Mat与.txt文件数据的读写操作