使用python语言递归遍历父文件夹下子目录中的图像并进行批量化操作的方法案例

需求分析

我们在做 CV 等任务时，会遇到对图像进行批量操作的情况，本篇文章以批量裁剪的案例，来探讨如何对父文件夹下各子文件夹中的图像遍历操作的方法。

代码实现

本案例是对'./img' 这个父文件夹下的子文件夹中的图像进行批量裁剪操作，文件夹目录结构如下（其中'./img' 为父文件夹，A、B…Z为父文件夹下的各个子文件夹）：

-<img>
   -<A>
      -图片.jpg
      -图片.jpg
      ...
      -图片.jpg
   -<B>
      -图片.jpg
      -图片.jpg
      ...
      -图片.jpg
   ...
   -<Z>
      -图片.jpg
      -图片.jpg
      ...
      -图片.jpg

实现代码如下¹：

import os
import shutil
import time
from time import sleep
from PIL import Image
from tqdm import tqdm


def makeDirs(path):
    # 创建新的输出子目录，若存在输出该目录则删除该目录及其子文件夹和文件
    if os.path.exists(path):
        shutil.rmtree(path)
    os.mkdir(path)


def resizePics(size):

    # default_path=os.getcwd()  ##设置默认打开目录，即当前目录
    # default_path = "/pics"

    # './img' 为即将进行遍历操作的父文件夹路径
    inpath = './img'  
    # outdir 变量为图像处理之后的输出文件夹
    outdir = os.path.basename(inpath) + '_resize'
    outpath = os.path.join(os.path.dirname(inpath), outdir)

    makeDirs(outpath)

    # 列出所在目录下的所有目录和文件
    lists = os.listdir(inpath)

    starttime = time.time()
    for i in tqdm(range(0, len(lists))):
        subdir = lists[i]
        subpath = inpath + "/" + subdir  # 子目录路径
        if os.path.isdir(subpath):
            # outsubdir = subdir + '-resize'
            outsubdir = subdir
            outsubpath = os.path.join(outpath, outsubdir)

            makeDirs(outsubpath)

            # 对文件夹下的照片文件调整大小
            # 列出某个子目录下的所有文件和目录
            flists = os.listdir(subpath)
            for j in tqdm(range(0, len(flists))):
                sleep(0.01)
                fname = flists[j]
                fpath = subpath + "/" + fname
                if os.path.isfile(fpath):
                    resizeSinglePic(fname, fpath, outsubpath)

    passtime = time.time()-starttime
    print('照片压缩完毕，总共花费了 %f s' % (passtime))


def resizeSinglePic(fname, fpath, outsubpath):
    fbasename = os.path.basename(fpath)
    fext = os.path.splitext(fpath)[-1]
    if fext in img_ext:
        img = Image.open(fpath)
        img.thumbnail(size)
        outfile = outsubpath + "/" + fbasename
        img.save(outfile)


img_ext = ['.bmp', '.jpeg', '.JPEG','.gif', '.psd', '.png', '.JPG', '.jpg']
size = (64, 64)  # 这行代码是本案例的需求实现，即将图像reshape为64*64尺寸
resizePics(size)

通过上述代码，即可实现对'./img' 父文件夹下A、B…Z这些子文件夹中图像的批量操作。

写到这里，差不多本文就要结束了。如果有问题可以在下方留言区留言交流。如果我的这篇文章帮助到了你，那我也会感到很高兴，一个人能走多远，在于与谁同行。

---

1. 使用python对文件夹下的照片进行批量修改尺寸 ↩︎