python/cv2对长方形图像进行等比例缩放及对周围常数值(0)填充zero padding

#coding:utf-8
import os
import cv2
import glob
import numpy as np

def mkdir(path):
    if not os.path.exists(path):
        os.mkdir(path)

def process_image(img):
    size = img.shape
    h, w = size[0], size[1]
    #长边缩放为min_side 
    scale = max(w, h) / float(min_side)
    new_w, new_h = int(w/scale), int(h/scale)
    resize_img = cv2.resize(img, (new_w, new_h))
    # 填充至min_side * min_side
    if new_w % 2 != 0 and new_h % 2 == 0:
        top, bottom, left, right = (min_side-new_h)/2, (min_side-new_h)/2, (min_side-new_w)/2 + 1, (min_side-new_w)/2
    elif new_h % 2 != 0 and new_w % 2 == 0:
        top, bottom, left, right = (min_side-new_h)/2 + 1, (min_side-new_h)/2, (min_side-new_w)/2, (min_side-new_w)/2
    elif new_h % 2 == 0 and new_w % 2 == 0:
        top, bottom, left, right = (min_side-new_h)/2, (min_side-new_h)/2, (min_side-new_w)/2, (min_side-new_w)/2
    else:
        top, bottom, left, right = (min_side-new_h)/2 + 1, (min_side-new_h)/2, (min_side-new_w)/2 + 1, (min_side-new_w)/2
    pad_img = cv2.copyMakeBorder(resize_img, top, bottom, left, right, cv2.BORDER_CONSTANT, value=[0,0,0]) #从图像边界向上,下,左,右扩的像素数目
    #print pad_img.shape
    #cv2.imwrite("after-" + os.path.basename(filename), pad_img)
    return pad_img

def main(data_dir):
    save_dir = data_dir + "_pad_" + str(min_side)
    mkdir(save_dir)
    num = 0
    for label in os.listdir(data_dir):
        num += 1
        print("%d/%d, label:%s" %(num, len(os.listdir(data_dir)), label))
        mkdir(os.path.join(save_dir, label))
        for img in glob.glob(os.path.join(data_dir, label, "*.jpg")):
            #print img
            image = cv2.imread(img)
            if type(image) == type(None):
                print("damaged image %s, del it" %(img))
                os.remove(img)
                continue
            img_pad = process_image(image)
            cv2.imwrite(os.path.join(save_dir, label, os.path.basename(img)), img_pad)


if __name__ == "__main__":
    min_side = 224
    #min_side = 256
    main(data_dir = "val")
    main(data_dir = "train")

原图：

等比例缩放，并补0填充后,224*224：

原图：

处理后：