File size: 1,422 Bytes
3ccac06 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 | 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)
|