Эссе - Пакетное изменение размера изображений Python

Python

Эссе - Пакетное изменение размера изображений Python

Справочная документация:

Мой собственный набор данных беспорядочный, и размер картинок разный.Вы можете подогнать размер картинок к единому размеру с помощью пакетов Python, что удобно для обучения.

import os
from PIL import Image


def resize(old_path, new_path, size, resample):
    """ 
     通过指定的resample方式调整old_path下所有的jpg图片为指定的size，并保存到new_path下 
    """
    if os.path.isdir(old_path):
        for child in os.listdir(old_path):

            if child.find('.jpg') > 0:
                im = Image.open(old_path + child)
                im_resized = im.resize(size=size, resample=resample)
                if not os.path.exists(new_path):
                    os.makedirs(new_path)
                print(child, 'resize successfully!')
                im_resized.save(new_path + child, im.format)
            child_path = old_path + child + '/'

            resize(child_path, new_path, size, resample)


if __name__ == "__main__":
    old_path = './test_dataset/'
    new_path = './test_dataset_train/'
    size = (1280, 720)
    resample = Image.BILINEAR # 使用线性插值法重采样
    resize(old_path, new_path, size, resample)
Категории