需求分析
我们在做 CV 等任务时,会遇到对图像进行批量操作的情况,本篇文章以批量裁剪的案例,来探讨如何对父文件夹下各子文件夹中的图像遍历操作的方法。
代码实现
本案例是对'./img'
这个父文件夹下的子文件夹中的图像进行批量裁剪操作,文件夹目录结构如下(其中'./img'
为父文件夹,A
、B
…Z
为父文件夹下的各个子文件夹):
-<img>
-<A>
-图片.jpg
-图片.jpg
...
-图片.jpg
-<B>
-图片.jpg
-图片.jpg
...
-图片.jpg
...
-<Z>
-图片.jpg
-图片.jpg
...
-图片.jpg
实现代码如下1:
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
这些子文件夹中图像的批量操作。
写到这里,差不多本文就要结束了。如果有问题可以在下方留言区留言交流。如果我的这篇文章帮助到了你,那我也会感到很高兴,一个人能走多远,在于与谁同行。