Merge pull request #392 from KananMahammadli/master

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

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")