应学长要求写一个函数,实现:
1、对所有输入的图片进行等比例缩放,目标格式为200*64,要求至少有一边达到要求,不足的另一边用黑色补充
2、应格式缘故,只能用PIL库,不能用cv
3、输入图片大小不定,可能超过200*64,也可能不足
实现代码:
from PIL import Image
def pad_image(image, target_size):
"""
:param image: input image
:param target_size: a tuple (num,num)
:return: new image
"""
iw, ih = image.size # 原始图像的尺寸
w, h = target_size # 目标图像的尺寸
print("original size: ",(iw,ih))
print("new size: ", (w, h))
scale = min(w / iw, h / ih) # 转换的最小比例
# 保证长或宽,至少一个符合目标图像的尺寸 0.5保证四舍五入
nw = int(iw * scale+0.5)
nh = int(ih * scale+0.5)
print("now nums are: ", (nw, nh))
image = image.resize((nw, nh), Image.BICUBIC) # 更改图像尺寸,双立法插值效果很好
#image.show()
new_image = Image.new('RGB', target_size, (0, 0, 0)) # 生成黑色图像
# // 为整数除法,计算图像的位置
new_image.paste(image, ((w - nw) // 2, (h - nh) // 2)) # 将图像填充为中间图像,