Python图片批量处理二三事(二)
2017-02-19 本文已影响181人
mugichya
上一篇讲了文件重命名的批处理,现在进入到具体的文件(这里讲图片)操作。如果像前面那样无脑地拿到了一堆没有经过筛选的图片文件,而我们最终想要的是有统一标准的图片,比如一定的像素大小,那么也可以用强大的Image模块实现。
原始待整理文件夹 大小不一的图片模块介绍
首先从PIL库(需要安装)导入Image,对文件操作也少不了os,此外再介绍一个glob模块, 是Python最简单的模块之一,内容非常少,用它可以查找符合特定规则的文件路径名。
- os
在上一篇有介绍和友情链接,可以参看 - Image
里面有许多通用操作,实现对图片属性的修改 - glob
这里用到print glob.glob(path) 获取指定目录下的所有文件
实现
- 找到文件夹
导入库就可以开始了,首先找到我们所有需要处理的文件夹,用一个find_folder()函数:
def find_folder():
in_folder_list = []
folders = os.listdir(in_path)
for f in folders:
folder_path = in_path + f
in_folder_list.append(f)
image_pixel(folder_path) # 将得到的文件路径依次放入像素处理函数中
print('The number of input folders is:', len(in_folder_list)) # 统计一下待处理的文件数目
- 修改图片尺寸,高度一定等比例缩放
def image_pixel(folder_path):
for image in glob.glob(folder_path + '/内容*.jpg'): # 以前面修改过的文件夹为例
img_path, img_name = os.path.split(image)
filterame, exts = os.path.splitext(img_name)
# 输出路径
# out_folder_path = 'E:/XX/XX/test_out/'
# 如果不存在
if os.path.isdir(out_folder_path) is False:
os.mkdir(out_folder_path)
img = Image.open(image)
# img.show()
w, h = img.size
if h != 340:
k = 340 / h # 将图片高度转换为340,如果不是则得到比例系数进行等比例缩放
new_img = img.resize((int(w * k), int(h * k)), Image.BILINEAR)
out_img_path = out_folder_path + folder_path.split('/')[4] + '/'
if os.path.isdir(out_img_path) is False:
os.mkdir(out_img_path)
# print(out_img_path)
new_img.save(out_img_path + filterame + '.jpg')
- 运行,输出
if __name__ == '__main__':
find_folder()
out_folder = os.listdir(out_folder_path)
for f in out_folder:
out_folder_list.append(f)
print('The number of output folders is:', len(out_folder_list))
print('done')
运行完后可以统计下文件数目,然后“done”提示完成,如图所示的所有文件夹处理完成前后不过几分钟时间。