aamsko commited on
Commit
f3084c9
Β·
verified Β·
1 Parent(s): 4e95405

Create utils/image_utils.py

Browse files
Files changed (1) hide show
  1. utils/image_utils.py +9 -0
utils/image_utils.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ from PIL import Image
2
+ import numpy as np
3
+
4
+ def resize_image(image, max_size=1024):
5
+ w, h = image.size
6
+ if max(w, h) > max_size:
7
+ scale = max_size / max(w, h)
8
+ return image.resize((int(w * scale), int(h * scale)))
9
+ return image