利用python或C++读取指定文件夹下的所有图片

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接： https://blog.csdn.net/zhou4411781/article/details/95445913

1.python读取指定文件夹下的所有图片路径和图片文件名

import cv2
from os import walk,path

def get_fileNames(rootdir):
    data=[]
    prefix = []
    for root, dirs, files in walk(rootdir, topdown=True):
        for name in files:
            pre, ending = path.splitext(name)
            if ending != ".jpg" and ending != ".jepg" and ending != ".png":
                continue
            else:
                data.append(path.join(root, name))
                prefix.append(pre)
    return data, prefix



if __name__ == '__main__':

    images, preifx = get_fileNames('/home/yasin/target_pics') #得到指定文件夹下的图片，例如.jpg，.jepg或.png等，可根据上述代码更改
    for i in range(len(images)):
        img = cv2.imread(images[i])
        w = img.shape[1]
        l = img.shape[0]
        img_res = cv2.resize(img, (w*2,l*2)) #对图片操作后
        cv2.imwrite("./resized_wb/{}.jpg".format(preifx[i]),img_res)

2.C++得到指定文件夹下的所有图片并返回读取Mat值

#include <stdio.h>
#include <highgui.h>
#include <opencv2/opencv.hpp>
#include <cv.h>

using namespace std;
using namespace cv;


//读取指定文件下的所有图片
vector<Mat> read_images_in_folder(cv::String pattern)
{
	vector<cv::String> fn;
	glob(pattern, fn, false);

	vector<Mat> images;
	// vector<cv::String>&prefix  //
	size_t count = fn.size(); //number of png files in images folder
	for (size_t i = 0; i < count; i++)
	{
	    // prefix.push_back(fn[i].substr(20, 4)); // 此处可以得到文件名的子字符串，可以获取图片前缀
		images.push_back(imread(fn[i])); //直读取图片并返回Mat类型
		//imshow("img", imread(fn[i]));
		//waitKey(1000);
	}
	return images;
}

int main()
{

	cv::String pattern = "./*.jpg";

	//遍历得到目标文件中所有的.jpg文件
	vector<Mat> images = read_images_in_folder(pattern);

	for (int i = 0; i < images.size(); i++)
	{
		imshow("img", images[i]);
		waitKey(1000);
	}
	// system("pause");
}