c++ OpenCV 读取某个文件夹下的所有图片

原创已于 2022-06-20 11:47:05 修改 · 7k 阅读

12 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #OpenCV

于 2022-04-23 11:05:17 首次发布

OpenCV 专栏收录该内容

17 篇文章

订阅专栏

这段代码展示了两种方法利用OpenCV读取指定文件夹中所有.jpg图片。方案一通过`cv::glob`函数获取文件名列表，然后逐个加载图片；方案二同样使用`cv::glob`但增加了文件夹存在的检查。

方案一

#include <vector>
#include <opencv2/imgproc.hpp>
#include <opencv2/highgui.hpp>

std::vector<cv::Mat> ReadImage(cv::String pattern)
{
    std::vector<cv::String> fn;
    cv::glob(pattern, fn, false);
    std::vector<cv::Mat> images;
    int count = fn.size(); //number of png files in images folder
    for (int i = 0; i < count; i++)
    {
        images.emplace_back(cv::imread(fn[i]));
    }
    return images;
}

int main()
{
    cv::String pattern="/home/guopei/桌面/test_imgs/*.jpg";
    std::vector<cv::Mat> img_list = ReadImage(pattern);
    return 0;
}

方案二：

#include <iostream>
#include "opencv2/core.hpp"
#include "opencv2/imgproc.hpp"
#include "opencv2/highgui.hpp"
#include <opencv2/core/utils/filesystem.hpp>


int main() {
    std::string img_dir = "../imgs";
    if(cv::utils::fs::exists(img_dir))
    {
        std::cout << "该文件夹存在" << std::endl;
    }

    // 获取当前文件夹下指定格式的文件
    std::vector<cv::String> img_lists;
    cv::utils::fs::glob(img_dir, "*.jpg", img_lists);
    for(auto name:img_lists)
    {
        std::cout << name << std::endl;
    }
    
    return 0;
}