第 0005 题: 你有一个目录,装了很多照片,把它们的尺寸变成都不大于 iPhone5 分辨率的大小。
解答:本题分以下几步来完成:
(1)递归获取目录中的所有图片文件
(2)iPhone5的分辨率为1136*640,先获取图片与iPhone的长宽比,如长的比例大于宽的比例,则按照长进行修改,否则按照宽的比例进行修改
(3)使用PIL中的Image类
#encoding:utf-8
from PIL import Image
import os
# ihone5的分辨率为1136*640
# 递归文件获取所有的图片
image_list=[]
def get_image_list(path):
filelist = os.listdir(path)
for file in filelist:
file_path = os.path.join(path,file)
if os.path.isfile(file_path):
if file.endswith('.jpeg') or file.endswith('.jpg') or file.endswith('.png'):
image_list.append(file_path)
elif os.path.isdir(file_path):
get_image_list(file_path)
# 修改图片尺寸
def resize_image(path,image_list,length=1036,width=640):
for file in image_list:
dirname,filename = os.path.split(file)
save_path = dirname.replace(path,path+"_resize")
if not os.path.exists(save_path):
os.makedirs(save_path)
image = Image.open(file)
image_length = image.size[0]
image_width = image.size[1]
# 如果图片长的比例大于宽的比例,则按照长进行修改
if image_length > length and image_length/length > image_length/width:
new_image = image.resize((length, int(width / (image_length / length))))
new_image.save(save_path + os.path.sep + filename)
# 如果图片宽的比例大于长的比例,则按照宽的比例进行修改
elif image_width > width and image_length/length < image_length/width:
new_image = image.resize((int(length / (image_length / width)), width))
new_image.save(save_path+os.path.sep+filename)
else:
print("不需要修改图片尺寸")
image.save()
if __name__ == '__main__':
path = "D:\\training\\image"
get_image_list(path)
print(image_list)
resize_image(path,image_list)
测试结果: