padding resize,
在图像的基础上填充0像素来扩张图像,优点是不会改变原图像内容比例
import cv2
import numpy as np
# Load the image
img = cv2.imread('path/to/your/image.png', cv2.IMREAD_UNCHANGED)
#填充需要的新的高和宽
new_img = PaddingScale(img, h, w)
def PaddingScale(img,h=128,w=128):
# 获得图像高宽
height, width = img.shape[:2]
#缩放系数
scale_factor = min(h / height, w / width)
# 比较图像的高和宽是否达到需要的大小
if height < h and width < w:
# 计算需要padding多少像素
pad_height = (h - height) // 2
pad_width = (w - width) // 2
# zero padding
padded_img = cv2.copyMakeBorder(img, pad_height, pad_height, pad_width, pad_width, cv2.BORDER_CONSTANT,value=(0, 0, 0))
img = padded_img
else:
## 如果宽高更大,就直接缩放
#img=cv2.resize(img,(w,h),interpolation=cv2.INTER_CUBIC)
else:
# 图像已经更大了,先将图像等比例缩放到小于128,然后再zero padding到128
# 等比缩放
resized_img = cv2.resize(img, (0, 0), fx=scale_factor, fy=scale_factor)
# 获取填充的像素大小
pad_height = (128 - resized_img.shape[0] % 128) % 128
pad_width = (128 - resized_img.shape[1] % 128) % 128
# 将图像填充到128
padded_img = cv2.copyMakeBorder(resized_img, 0, pad_height, 0, pad_width, cv2.BORDER_CONSTANT, value=(0, 0, 0))
img=padded_img
return img