OpenCV图像编程实例系列-图像文件批量读取
2016-08-07 17:01 zhuwei1988 阅读(9507) 评论(1) 编辑 收藏 举报本博文摘录《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 // 作者：朱伟 zhu1988wei@163.com 3 // 来源：《OpenCV图像处理编程实例》 4 // 博客：http://blog.csdn.net/zhuwei1988 5 // 更新：2016-8-1 6 // 说明：版权所有，引用或摘录请联系作者，并按照上面格式注明出处，谢谢。// 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 // 作者：朱伟 zhu1988wei@163.com 3 // 来源：《OpenCV图像处理编程实例》 4 // 博客：http://blog.csdn.net/zhuwei1988 5 // 更新：2016-8-1 6 // 说明：版权所有，引用或摘录请联系作者，并按照上面格式注明出处，谢谢。// 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获取当前文件夹名的句柄；然后遍历当前文件夹名下的所有文件，将得到的所有文件名称转换后赋值于图像文件向量；最后遍历完当前文件下的所有文件，生成相应图像文件索引名称，用于文件夹中所有图像文件的读取，在读取单个图像文件后可进行相关的图像处理操作。
