save_images.py 文件源码

def resize_to_nearest_aspect_ratio(img, divide_base=4, resize_base=256):
    w, h = img.shape[0], img.shape[1]
    #print(w,h)
    if w < h:
        if resize_base == 0:
            resize_base = w - w % divide_base
        s0 = resize_base
        s1 = int(h * resize_base / w)
        s1 = s1 - s1 % divide_base
    else:
        if resize_base == 0:
            resize_base = h - h % divide_base
        s1 = resize_base
        s0 = int(w * resize_base / h)
        s0 = s0 - s0 % divide_base
    #print(s1,s0)
    return cv2.resize(img, (s1, s0), interpolation=cv2.INTER_AREA)


# Input imgs format: (batch, channels, width, height)