思路:
- 首先,把图像填充为正方形(以图像的最长边为正方形新图边长,对较短边长居中填充空白)
- 循环迭代切分图像
- 保存生成的图像列表
from PIL import Image
import sys
import os
# 将图片填充为正方形
def fill_image(image):
width, height = image.size
# 选取长和宽中较大值作为新图片的
new_image_length = width if width > height else height
# 生成新图片[白底]
new_image = Image.new(image.mode, (new_image_length, new_image_length), color='white')
# 将之前的图粘贴在新图上,居中
if width > height: # 原图宽大于高,则填充图片的竖直维度
# (x,y)二元组表示粘贴上图相对下图的起始位置
new_image.paste(image, (0, int((new_image_length - height) / 2)))
else:
new_image.paste(image, (int((new_image_length - width) / 2), 0))
return new_image
# 平均切四份图
def cut_image(image):
width, height = image.size
item_width = int(width / 2)
box_list = []
# (left, upper, right, lower)
for i in range(0, 2): # 两重循环,生成4张图片基于原图的位置
for j in range(0, 2):
# print((i*item_width,j*item_width,(i+1)*item_width,(j+1)*item_width))
box = (j * item_width, i * item_width, (j + 1) * item_width, (i + 1) * item_width)
box_list.append(box)
image_list = [image.crop(box) for box in box_list]
return image_list
# 保存
def save_images(image_list, saveDir, name): # image_list:图像数组 , savePath: 图像保存目录, name:新的图像名
index = 1
for image in image_list:
image.save(os.path.join(saveDir,str(index)+name))
index += 1
if __name__ == '__main__':
sourcepath = 'E:/图像集/图像集'
dirPath = 'img'
for sampledir in os.listdir(sourcepath):
# valClassNum += 1
imageDir = sourcepath + "/" + sampledir
dirPath = dirPath + '/' + sampledir
for name in os.listdir(imageDir):
file_path = imageDir + "/" + name
image = Image.open(file_path)
# image.show()
image = fill_image(image)
image_list = cut_image(image)
print(dirPath)
save_images(image_list, dirPath, name)
dirPath = 'img'