欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

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

程序员文章站 2022-04-09 12:33:27
本文实例为大家分享了python或c++读取指定文件夹下的所有图片,供大家参考,具体内容如下 1.python读取指定文件夹下的所有图片路径和图片文件名 imp...

本文实例为大家分享了python或c++读取指定文件夹下的所有图片,供大家参考,具体内容如下

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");
}

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。