Python – 使用letter box方法缩放图片，防止图片缩放时失真

1 使用letter box方法缩放图片，防止图片缩放时失真

假如我们有一张1920x1080的图片想缩放到224x224，如果不进行任何处理，那么图片肯定会出现失真的问题。如果要把一张920x1080的图片缩放到224x224，我们需要首先保持原始图片的纵横比，然后在需要补齐的部分使用一个默认值填充，这里选择的是128。

具体的示例代码如下：

1.1 使用PIL库

from PIL import Image

def resize_image_with_scale(image, size):
    iw, ih = image.size
    w, h = size
    scale = min(w/iw, h/ih)
    nw = int(iw*scale)
    nh = int(ih*scale)

    image = image.resize((nw, nh), Image.Resampling.BICUBIC)
    new_image = Image.new('RGB', size, (128, 128, 128))
    new_image.paste(image, ((w-nw)//2, (h-nh)//2))
    return new_image

if __name__ == '__main__':
    jpg_file_path = "C:\\Users\\xxx\\Desktop\\a.jpeg"
    img = Image.open(jpg_file_path)

    new_image = resize_image_with_scale(img, size=(224, 224))

    new_image.save("b.jpg")

图片处理前后的效果如下：

从结果上看，在两边的区域padding了(128,128,128)的这种纯灰度的值。这样就可以保持图片既可以缩放也可以不失真了。

1.2 使用opencv-python库

import cv2
import numpy as np

def resize_image_with_scale(image, size):
    # 获取原始图像的尺寸
    ih, iw = image.shape[:2]
    w, h = size

    # 计算缩放比例
    scale = min(w / iw, h / ih)
    nw = int(iw * scale)
    nh = int(ih * scale)

    # 使用OpenCV的resize函数调整图像大小
    resized_image = cv2.resize(image, (nw, nh), interpolation=cv2.INTER_CUBIC)

    # 创建一个新的图像，背景色为灰色（128, 128, 128）
    new_image = np.full((h, w, 3), (128, 128, 128), dtype=np.uint8)

    # 计算粘贴位置
    x = (w - nw) // 2
    y = (h - nh) // 2

    # 将调整大小后的图像粘贴到新图像上
    new_image[y:y + nh, x:x + nw] = resized_image

    # OpenCV默认使用BGR格式，如果你需要RGB格式，可以转换
    # new_image = cv2.cvtColor(new_image, cv2.COLOR_BGR2RGB)

    return new_image

if __name__ == '__main__':
    image_path = 'test.jpg'
    image = cv2.imread(image_path)
    resized_img = resize_image_with_scale(image, (640, 640))

    cv2.imshow('Resized Image', resized_img)
    cv2.waitKey(0)
    cv2.destroyAllWindows()