彩色图像长短边等比例缩放并填零补成正方形

最新推荐文章于 2024-02-16 22:49:17 发布

原创

最新推荐文章于 2024-02-16 22:49:17 发布 · 898 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#opencv #计算机视觉

该博客介绍了一种图像处理方法，将彩色图像按比例缩放成长宽比为1的正方形，长边缩放至固定尺寸，短边等比例缩放后用零填充。使用OpenCV进行通道分离与合并，并通过copyMakeBorder实现填充。代码实现了批量处理图片，将处理后的图像保存至指定文件夹。

（适用于彩色图）

图像长短边等比例缩放并填零补成正方形说明：将图像长边缩放至固定尺寸，并将短边等比例缩放，空出部分两边对称填零，补成正方形。

以下是由于opencv读取和保存的通道顺序默认指的不一样，所以分隔通道后再融合。

pil_image = cv2.imread(path_image)
b, g, r = cv2.split(pil_image)  # 通道分离,再重新合并操作
pil_image = cv2.merge([r, g, b])
output_image = img_pad(pil_image)
output_image.save(path_output)

完整代码如下：

import os
import numpy as np
from PIL import Image
import cv2

def img_pad(pil_file):
    w, h,c= pil_file.shape
    fixed_size = 139  # 输出正方形图片的尺寸

    if h >= w:
        factor = h / float(fixed_size)
        new_w = int(w / factor)
        if new_w % 2 != 0:
            new_w -= 1
        pil_file = cv2.resize(pil_file,(new_w, fixed_size))
        pad_w = int((fixed_size - new_w) / 2)
        array_file = np.array(pil_file)
        # array_file = np.pad(array_file, ((0, 0), (pad_w, fixed_size-pad_w)), 'constant')
        array_file = cv2.copyMakeBorder(array_file, 0, 0, pad_w, fixed_s