github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 20:15:21 +00:00
Code Issues Packages Projects Releases Wiki Activity

load img from web sorted

Browse Source
This commit is contained in:
Sefik Ilkin Serengil 2024-02-16 13:41:46 +00:00
parent aa18fa3976
commit d16f0765fa
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
deepface/modules/preprocessing.py
View File

@ -6,7 +6,6 @@ from pathlib import Path
# 3rd party # 3rd party
import numpy as np import numpy as np
import cv2 import cv2
from PIL import Image
import requests import requests
@ -36,11 +35,7 @@ def load_image(img: Union[str, np.ndarray]) -> Tuple[np.ndarray, str]:
# The image is a url # The image is a url
if img.startswith("http"): if img.startswith("http"):
return ( return load_image_from_web(url=img), img
np.array(Image.open(requests.get(img, stream=True, timeout=60).raw).convert("BGR")),
# return url as image name
img,
)
# The image is a path # The image is a path
if os.path.isfile(img) is not True: if os.path.isfile(img) is not True:
@ -57,6 +52,21 @@ def load_image(img: Union[str, np.ndarray]) -> Tuple[np.ndarray, str]:
return img_obj_bgr, img return img_obj_bgr, img
def load_image_from_web(url: str) -> np.ndarray:
"""
Loading an image from web
Args:
url: link for the image
Returns:
img (np.ndarray): equivalent to pre-loaded image from opencv (BGR format)
"""
response = requests.get(url, stream=True, timeout=60)
response.raise_for_status()
image_array = np.asarray(bytearray(response.raw.read()), dtype=np.uint8)
image = cv2.imdecode(image_array, cv2.IMREAD_COLOR)
return image
def load_base64(uri: str) -> np.ndarray: def load_base64(uri: str) -> np.ndarray:
"""Load image from base64 string. """Load image from base64 string.