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

14
deepface/commons/functions.py
View File

@ -4,6 +4,8 @@ import pandas as pd
import cv2 import cv2
import base64 import base64
from pathlib import Path from pathlib import Path
from PIL import Image
import requests
from deepface.detectors import FaceDetector from deepface.detectors import FaceDetector
@ -64,7 +66,6 @@ def loadBase64Img(uri):
return img return img
def load_image(img): def load_image(img):
exact_image = False exact_image = False
if type(img).__module__ == np.__name__: if type(img).__module__ == np.__name__:
exact_image = True exact_image = True
@ -73,10 +74,17 @@ def load_image(img):
if len(img) > 11 and img[0:11] == "data:image/": if len(img) > 11 and img[0:11] == "data:image/":
base64_img = True base64_img = True
url_img = False
if len(img) > 11 and img.startswith("http"):
url_img = True
#--------------------------- #---------------------------
if base64_img == True: if base64_img == True:
img = loadBase64Img(img) 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 elif exact_image != True: #image path passed as input
if os.path.isfile(img) != True: if os.path.isfile(img) != True:
@ -112,9 +120,9 @@ def detect_face(img, detector_backend = 'opencv', grayscale = False, enforce_det
else: else:
if detected_face == None: if detected_face == None:
if enforce_detection != True: if enforce_detection != True:
return img, img_region return img, img_region
else: else:
raise ValueError("Face could not be detected. Please confirm that the picture is a face photo or consider to set enforce_detection param to False.") raise ValueError("Face could not be detected. Please confirm that the picture is a face photo or consider to set enforce_detection param to False.")
def normalize_input(img, normalization = 'base'): def normalize_input(img, normalization = 'base'):