mirror of
https://github.com/serengil/deepface.git
synced 2025-06-07 20:15:21 +00:00
stream bug
This commit is contained in:
parent
d6baa8381c
commit
67149c629b
@ -85,7 +85,7 @@ def analysis(db_path, model_name, distance_metric, enable_face_analysis = True
|
||||
employee = employees[index]
|
||||
pbar.set_description("Finding embedding for %s" % (employee.split("/")[-1]))
|
||||
embedding = []
|
||||
img = functions.preprocess_face(img = employee, target_size = (input_shape_y, input_shape_x), enforce_detection = False)
|
||||
img = functions.preprocess_face(img = employee, target_size = (input_shape_y, input_shape_x), enforce_detection = False, detector_backend = 'opencv')
|
||||
img_representation = model.predict(img)[0,:]
|
||||
|
||||
embedding.append(employee)
|
||||
@ -196,7 +196,7 @@ def analysis(db_path, model_name, distance_metric, enable_face_analysis = True
|
||||
|
||||
if enable_face_analysis == True:
|
||||
|
||||
gray_img = functions.preprocess_face(img = custom_face, target_size = (48, 48), grayscale = True, enforce_detection = False)
|
||||
gray_img = functions.preprocess_face(img = custom_face, target_size = (48, 48), grayscale = True, enforce_detection = False, detector_backend = 'opencv')
|
||||
emotion_labels = ['Angry', 'Disgust', 'Fear', 'Happy', 'Sad', 'Surprise', 'Neutral']
|
||||
emotion_predictions = emotion_model.predict(gray_img)[0,:]
|
||||
sum_of_predictions = emotion_predictions.sum()
|
||||
@ -274,7 +274,7 @@ def analysis(db_path, model_name, distance_metric, enable_face_analysis = True
|
||||
|
||||
#-------------------------------
|
||||
|
||||
face_224 = functions.preprocess_face(img = custom_face, target_size = (224, 224), grayscale = False, enforce_detection = False)
|
||||
face_224 = functions.preprocess_face(img = custom_face, target_size = (224, 224), grayscale = False, enforce_detection = False, detector_backend = 'opencv')
|
||||
|
||||
age_predictions = age_model.predict(face_224)[0,:]
|
||||
apparent_age = Age.findApparentAge(age_predictions)
|
||||
@ -329,7 +329,7 @@ def analysis(db_path, model_name, distance_metric, enable_face_analysis = True
|
||||
#-------------------------------
|
||||
#face recognition
|
||||
|
||||
custom_face = functions.preprocess_face(img = custom_face, target_size = (input_shape_y, input_shape_x), enforce_detection = False)
|
||||
custom_face = functions.preprocess_face(img = custom_face, target_size = (input_shape_y, input_shape_x), enforce_detection = False, detector_backend = 'opencv')
|
||||
|
||||
#check preprocess_face function handled
|
||||
if custom_face.shape[1:3] == input_shape:
|
||||
|
Loading…
x
Reference in New Issue
Block a user