蓉蓉的计算机视觉打怪之路

Python opencv resize图片并保存原有的图像比例

2019-04-18  本文已影响0人  涂山容容

现有的图像是高瘦高瘦的,所以直接resize成矩形不合适。改变了整个结构。
所以采用的是先resize再padding的方式。
1.resize图片,先计算最长边的resize的比例,然后按照该比例resize。
2.计算四个边需要padding的像素宽度,然后padding。

def resize_img_keep_ratio(img_name,target_size):
    img = cv2.imread(img_name)
    old_size= img.shape[0:2]
    #ratio = min(float(target_size)/(old_size))
    ratio = min(float(target_size[i])/(old_size[i]) for i in range(len(old_size)))
    new_size = tuple([int(i*ratio) for i in old_size])
    img = cv2.resize(img,(new_size[1], new_size[0]))
    pad_w = target_size[1] - new_size[1]
    pad_h = target_size[0] - new_size[0]
    top,bottom = pad_h//2, pad_h-(pad_h//2)
    left,right = pad_w//2, pad_w -(pad_w//2)
    img_new = cv2.copyMakeBorder(img,top,bottom,left,right,cv2.BORDER_CONSTANT,None,(0,0,0))
    return img_new
if__name__=='main':
    img = ''
    target_size=[448,112]
    resize_img_keep_ratio()
上一篇下一篇

猜你喜欢

热点阅读