OpenCV圖像編程實例系列-圖像文件批量讀取

本博文摘錄《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所示。

 

// 功能：代碼 2-32 文件名連續情況下
// 作者：朱偉 [e-mail]
// 來源：《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 文件名無規則情況讀取
// 作者：朱偉 [e-mail]
// 來源：《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獲取當前文件夾名的句柄；然后遍歷當前文件夾名下的所有文件，將得到的所有文件名稱轉換后賦值於圖像文件向量；最后遍歷完當前文件下的所有文件，生成相應圖像文件索引名稱，用於文件夾中所有圖像文件的讀取，在讀取單個圖像文件后可進行相關的圖像處理操作。