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

Merge pull request #392 from KananMahammadli/master

Browse Source 
Modified functions.py so that it can read from https
This commit is contained in:
Sefik Ilkin Serengil 2021-12-23 09:51:20 +03:00 committed by GitHub
commit 5cfc94e5bf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
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 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,11 +74,18 @@ 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:
raise ValueError("Confirm that ",img," exists") raise ValueError("Confirm that ",img," exists")