数据增强:padding

2022-01-14  本文已影响0人  小小杨树

数据增强,使用padding进行数据增可以有效避免数据失真

"""数据增强,使用padding进行数据增可以有效避免数据失真"""

from PIL import Image


def letterbox_image(image, size):
    # 对图片进行resize,使图片不失真。在空缺的地方进行padding
    iw, ih = image.size
    w, h = size
    scale = min(w / iw, h / ih)
    nw = int(iw * scale)
    nh = int(ih * scale)

    image = image.resize((nw, nh), Image.BICUBIC)
    new_image = Image.new('RGB', size, (128, 128, 128))
    new_image.paste(image, ((w - nw) // 2, (h - nh) // 2))
    return new_image


img = Image.open("112.jpg")
new_image = letterbox_image(img, [416, 416])
new_image.show()

上一篇 下一篇

猜你喜欢

热点阅读