*批量生成多尺寸图像

import argparse
import glob
import os
from PIL import Image


def main():
    # 修改图像尺寸: 0.25为宽高都缩小至四分之一,4为宽高都放大4倍
    scale = 0.25
    # 输入文件路径
    input = 'C:\\Users\\mayuhua\\Desktop\\深度学习SR\\DF2K\\DIV2K_train_HR'
    # 输出文件路径
    output = 'C:\\Users\\mayuhua\\Desktop\\深度学习SR\\DF2K\\DIV2K_train_LR-0.25'

    path_list = sorted(glob.glob(os.path.join(input, '*')))
    print(path_list)
    for path in path_list:
        print(path)
        print(os.path.basename(path))
        basename = os.path.splitext(os.path.basename(path))[0]
        print(basename)


        img = Image.open(path)
        width, height = img.size

        rlt = img.resize((int(width * scale), int(height * scale)), resample=Image.LANCZOS)
        rlt.save(os.path.join(output, f'{basename}T.png'))



if __name__ == '__main__':
    """Generate multi-scale versions for GT images with LANCZOS resampling.
    """
    main()

你可能感兴趣的:(深度学习pytorch系列文章,深度学习)