Create utils/image_utils.py
Browse files- 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
|