Update src/util/image.py

src/util/image.py

@@ -11,3 +11,12 @@ def load_image_url(url, required_size = (224,224), image_type = 'array'):
     if image_type == 'array':
         img = (np.expand_dims(np.array(img), 0)/255).astype(np.float32)
     return img
+
+def load_image_path(path, required_size = (224,224), image_type = 'array'):
+    img = Image.open(path)
+    img = Image.fromarray(np.array(img))
+    if required_size is not None:
+        img = img.resize(required_size)
+    if image_type == 'array':
+        img = (np.expand_dims(np.array(img), 0)/255).astype(np.float32)
+    return img