mirror of
https://github.com/serengil/deepface.git
synced 2025-06-06 19:45:21 +00:00
Update functions.py
This commit is contained in:
parent
06ca8161bc
commit
6522fa7d02
@ -102,13 +102,16 @@ def load_image(img):
|
||||
# The image is a path
|
||||
if os.path.isfile(img) is not True:
|
||||
raise ValueError(f"Confirm that {img} exists")
|
||||
|
||||
with open(img, "rb") as img:
|
||||
chunk = img.read()
|
||||
|
||||
# For reading images with unicode names
|
||||
with open(img, "rb") as img_f:
|
||||
chunk = img_f.read()
|
||||
chunk_arr = np.frombuffer(chunk, dtype=np.uint8)
|
||||
img = cv2.imdecode(chunk_arr, cv2.IMREAD_COLOR)
|
||||
|
||||
return img
|
||||
# This causes troubles when reading files with non english names
|
||||
# return cv2.imread(img)
|
||||
|
||||
|
||||
# --------------------------------------------------
|
||||
|
Loading…
x
Reference in New Issue
Block a user