Advertisement
Guest User

Untitled

a guest
Dec 12th, 2020
1,885
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
Python 0.94 KB | None | 0 0
  1. from PIL import Image
  2.  
  3. def make_square(im, min_size=256, fill_color=(255, 255, 255)):
  4.     x, y = im.size
  5.     size = max(min_size, x, y)
  6.     new_im = Image.new('RGBA', (size, size), fill_color)
  7.     new_im.paste(im, (int((size - x) / 2), int((size - y) / 2)))
  8.     return new_im
  9.  
  10. import os
  11. import shutil
  12.  
  13. formats = ('jpg', 'jpeg', 'png', 'jtif', 'webp')
  14. folder = 'D:/picures/' #root folder
  15. src = folder+'photos' #pictures to rectify
  16. dest = folder+'photos_rect' #rectified picture folder
  17.  
  18. for root, dirs, files in os.walk(src):
  19.     for file in files:
  20.         if file.lower().endswith(formats):
  21.             DEST = f"{dest}/{root[3:]}"
  22.             if not os.path.exists(DEST):
  23.                 os.makedirs(DEST)
  24.             shutil.copy(os.path.join(root, file), os.path.join(DEST, file))
  25.  
  26.  
  27. for root, dirs, files in os.walk(dest):
  28.     for file in files:
  29.         path = os.path.join(root, file)
  30.         im = Image.open(path)
  31.         im = make_square(im)
  32.         im = im.convert('RGB')
  33.         im.save(path, 'JPEG')      
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement