github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-12 22:37:09 +00:00
Code Issues Packages Projects Releases Wiki Activity

Modified functions.py so that it can read from https

Browse Source
This commit is contained in:
Kanan_Mahammadli 2021-12-22 19:09:15 +03:00
parent f877591abc
commit 799eb6c7a0
1 changed files with 11 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
deepface/commons/functions.py
View File

@ -4,6 +4,8 @@ import pandas as pd
import cv2
import base64
from pathlib import Path
from PIL import Image
import requests
from deepface.detectors import FaceDetector
@ -64,7 +66,6 @@ def loadBase64Img(uri):
return img
def load_image(img):
exact_image = False
if type(img).__module__ == np.__name__:
exact_image = True
@ -73,11 +74,18 @@ def load_image(img):
if len(img) > 11 and img[0:11] == "data:image/":
base64_img = True
url_img = False
if len(img) > 11 and img.startswith("http"):
url_img = True
#---------------------------
if base64_img == True:
img = loadBase64Img(img)
elif url_img:
img = np.array(Image.open(requests.get(img, stream=True).raw))
elif exact_image != True: #image path passed as input
if os.path.isfile(img) != True:
raise ValueError("Confirm that ",img," exists")