批量读取并保存图片

批量读取并保存图片

针对文件夹里无规则命名的图片

import os
import cv2


def getFileList(dir, Filelist, ext=None):
    """
    获取文件夹及其子文件夹中文件列表
    输入 dir：文件夹根目录
    输入 ext: 扩展名
    返回： 文件路径列表
    """
    newDir = dir
    if os.path.isfile(dir):
        if ext is None:
            Filelist.append(dir)
        else:
            if ext in dir[-3:]:
                Filelist.append(dir)

    elif os.path.isdir(dir):
        for s in os.listdir(dir):
            newDir = os.path.join(dir, s)
            getFileList(newDir, Filelist, ext)

    return Filelist
org_img_folder = 'left_1'  #选择图片所在文件夹

    # 检索文件
    imglist = getFileList(org_img_folder, [], 'jpg')
    print('本次执行检索到 ' + str(len(imglist)) + ' 张图像\n')

    i=1
    for imgpath in imglist:
     	# 对每幅图像执行相关操作，如
        img = cv2.resize(imgpath,(600,600))
        #按照1,2，3....命名处理后的图片并保存在left_1_undistort文件夹下   
        cv2.imwrite('left_1_undistort/'+str(i)+'.jpg', img)
        i+=1