preprocess.py (776B)
1 #!/usr/bin/env python3 2 3 from os import listdir, mkdir 4 from os.path import isdir, isfile, join 5 from PIL import Image 6 from shutil import rmtree 7 8 from src.common import HEIGHT, WIDTH 9 10 11 BASE_DIR = './assets' 12 OUT_DIR = join(BASE_DIR, 'out') 13 14 15 def main(): 16 if isdir(OUT_DIR): 17 rmtree(OUT_DIR) 18 19 mkdir(OUT_DIR) 20 21 for fpath in (join(BASE_DIR, f) for f in listdir(BASE_DIR) if isfile(join(BASE_DIR, f))): 22 fname = fpath.rsplit('/', 1)[-1].rsplit('.', 1)[0] 23 24 print(f'Fpath: {fpath}, Fname: {fname}') 25 print(f'Target size: {WIDTH}x{HEIGHT}') 26 27 image = Image.open(fpath) 28 29 resized = image.resize((WIDTH, HEIGHT)) 30 31 resized.save(join(OUT_DIR, f'{fname}.gif'), format='GIF', save_all=True) 32 33 34 if __name__ == '__main__': 35 main() 36