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獲取當前文件夾名的句柄;然后遍歷當前文件夾名下的所有文件,將得到的所有文件名稱轉換后賦值于圖像文件向量;最后遍歷完當前文件下的所有文件,生成相應圖像文件索引名稱,用于文件夾中所有圖像文件的讀取,在讀取單個圖像文件后可進行相關的圖像處理操作。
免責聲明:本站發布的內容(圖片、視頻和文字)以原創、轉載和分享為主,文章觀點不代表本網站立場,如果涉及侵權請聯系站長郵箱:is@yisu.com進行舉報,并提供相關證據,一經查實,將立刻刪除涉嫌侵權內容。