本博文摘录《OpenCV图像处理编程实例》2.4章节,更详细的内容请参考本书。图书购买地址: 当当:http://product.dangdang.com/23956649.html 京东:http://item.jd.com/11929148.html
2.4.6 图像批量读取——规则
在进行图片序列处理时,我们常常需要读取文件夹下的每一个图片,然后再进行分析处理,因此需要对文件名连续及无规则情况分开讨论。对于文件名连续的情况,文件读取就简单得多,可以利用sprintf函数实现在窗口中连续读取同一文件夹下的图片序列,而对于无规则的情况则可以采用基于VC下WIN32_ FIND_DATA文件的读取方式。文件名连续情况下的读取如代码2-32所示。
1 // 功能:代码 2-32 文件名连续情况下
2 // 作者:朱伟
7 #include <iostream>
8 #include <stdio.h>
9 #include <stdlib.h>
10 #include <opencv2/highgui/highgui.hpp>
11 #include <opencv2/imgproc/imgproc.hpp>
12 using namespace cv;
13 using namespace std;
14 int main()
15 {
16 // 定义相关参数
17 const int num = 4;
18 char fileName[50];
19 char windowName[50];
20 cv::Mat srcImage;
21 for (int i = 1; i <= num; i++)
22 {
23 // sprintf读入指定路径下图片序列
24 sprintf_s(fileName, "..\\images\\test\\1 (%d).jpg", i);
25 sprintf_s(windowName, "NO%d", i);
26 // 按照图像文件名读取
27 srcImage = cv::imread(fileName);
28 if (!srcImage.data)
29 {
30 std::cout << "No data!" << std::endl;
31 return -1;
32 }
33 cv::namedWindow(windowName);
34 cv::imshow(windowName, srcImage);
35 std::cout << "NO: " << i << std::endl;
36 //cv::waitKey(0);
37 /* 该处可以添加处理步骤 */
38 }
39 cv::waitKey(0);
40 return 0;
41 }
第16行代码利用sprintf将对应的图像文件路径转换为char*,在这种文件名连续的时候可以选中文件夹中的所有图像文件,然后用鼠标右键选定并重命名,键入1后,文件夹的所有文件自动命名为1 (k).jpg,其中k取值为1,2……然后就可以根据本方法进行批量读取了。
2.4.7 图像批量读取——无规则
文件名无规则的情况读取如代码2-33所示。
1 // 功能:代码 2-33 文件名无规则情况读取
2 // 作者:朱伟
7 #include <opencv2/core/core.hpp>
8 #include <opencv2/highgui/highgui.hpp>
9 #include <opencv2/imgproc/imgproc.hpp>
10 #include <iostream>
11 #include <stdio.h>
12 #include <windows.h>
13 using namespace std;
14 // LPCWSTR转string
15 std::string WChar2Ansi(LPCWSTR pwszSrc)
16 {
17 int nLen = WideCharToMultiByte(CP_ACP, 0, pwszSrc, -1, NULL, 0, NULL, NULL);
18
19 if (nLen <= 0) return std::string("");
20
21 char* pszDst = new char[nLen];
22 if (NULL == pszDst) return std::string("");
23
24 WideCharToMultiByte(CP_ACP, 0, pwszSrc, -1, pszDst, nLen, NULL, NULL);
25 pszDst[nLen - 1] = 0;
26
27 std::string strTemp(pszDst);
28 delete[] pszDst;
29
30 return strTemp;
31 }
32
33 // 利用winWIN32_FIND_DATA读取文件下的文件名
34 void readImgNamefromFile(char* fileName, vector <string> &imgNames)
35 {
36 // vector清零 参数设置
37 imgNames.clear();
38 WIN32_FIND_DATA file;
39 int i = 0;
40 char tempFilePath[MAX_PATH + 1];
41 char tempFileName[50];
42 // 转换输入文件名
43 sprintf_s(tempFilePath, "%s/*", fileName);
44 // 多字节转换
45 WCHAR wstr[MAX_PATH] = { 0 };
46 MultiByteToWideChar(CP_ACP, 0, tempFilePath, -1, wstr, sizeof(wstr));
47 // 查找该文件待操作文件的相关属性读取到WIN32_FIND_DATA
48 HANDLE handle = FindFirstFile(wstr, &file);
49 if (handle != INVALID_HANDLE_VALUE)
50 {
51 FindNextFile(handle, &file);
52 FindNextFile(handle, &file);
53 // 循环遍历得到文件夹的所有文件名
54 do
55 {
56 sprintf(tempFileName, "%s", fileName);
57 imgNames.push_back(WChar2Ansi(file.cFileName));
58 imgNames[i].insert(0, tempFileName);
59 i++;
60 } while (FindNextFile(handle, &file));
61 }
62 FindClose(handle);
63 }
64 int main()
65 {
66 // 设置读入图像序列文件夹的路径
67 char* fileName = "..\\images\\test\\";
68 std::vector <string> imgNames;
69 // 获取对应文件夹下所有文件名
70 readImgNamefromFile(fileName, imgNames);
71 // 遍历对应文件夹下所有文件名
72 for (int i = 0; i < imgNames.size(); i++)
73 {
74 cv::Mat img = cv::imread(imgNames[i]);
75 if (!img.data)
76 return -1;
77 /* 可添加图像处理算法code*/
78 cv::imshow("im", img);
79 cv::waitKey(0);
80 }
81 return 0;
82 }
利用winWIN32_FIND_DATA读取文件夹下文件的思路:首先转换文件夹名,利用FindFirstFile获取当前文件夹名的句柄;然后遍历当前文件夹名下的所有文件,将得到的所有文件名称转换后赋值于图像文件向量;最后遍历完当前文件下的所有文件,生成相应图像文件索引名称,用于文件夹中所有图像文件的读取,在读取单个图像文件后可进行相关的图像处理操作。