| import argparse |
| import glob |
| import os |
| from PIL import Image |
|
|
| def main(args): |
| scale_list = [0.75, 0.5, 0.33] |
| shortestEdge = 240 |
| path_list = sorted(glob.glob(os.path.join(args.input, '*'))) |
| for path in path_list: |
| print(path) |
| basename = os.path.splitext(os.path.basename(path))[0] |
|
|
| img = Image.open(path) |
| width, height = img.size |
| for idx, scale in enumerate(scale_list): |
| print(f'\t{scale:.2f}') |
| rlt = img.resize((int(width*scale), int(height*scale)), resample=Image.LANCZOS) |
| rlt.save(os.path.join(args.output, f'{basename}T{idx}.png')) |
| |
| if width < height: |
| ratio = height / width |
| width = shortestEdge |
| height = int(width*ratio) |
| else: |
| ratio = width/height |
| height = shortestEdge |
| width = int(height*ratio) |
| rlt = img.resize((int(width), int(height)), resample=Image.LANCZOS) |
| rlt.save(os.path.join(args.output, f'{basename}T{idx+1}.png')) |
|
|
| if __name__ == "__main__": |
| parser = argparse.ArgumentParser() |
| parser.add_argument('--input', type=str, default='dataset\Brain Tumor\Brain Tumor', help='Input Folder') |
| parser.add_argument('--output', type=str, default='dataset\Brain Tumor\Brain_Tumor_multiscale', help='Output Folder') |
| args = parser.parse_args() |
|
|
| os.makedirs(args.output, exist_ok=True) |
| main(args) |
|
|
