github/deepface
1
1
Fork 0
mirror of https://github.com/serengil/deepface.git synced 2025-06-07 12:05:22 +00:00
Code Issues Packages Projects Releases Wiki Activity

[update] add emotions batch test

Browse Source
This commit is contained in:
Nat Lee 2025-01-20 18:14:48 +08:00
parent 0971fcde1f
commit db4b749c98
1 changed files with 12 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
tests/test_analyze.py
View File

@ -1,5 +1,6 @@
# 3rd party dependencies # 3rd party dependencies
import cv2 import cv2
import numpy as np
# project dependencies # project dependencies
from deepface import DeepFace from deepface import DeepFace
@ -136,7 +137,7 @@ def test_analyze_for_different_detectors():
else: else:
assert result["gender"]["Man"] < result["gender"]["Woman"] assert result["gender"]["Man"] < result["gender"]["Woman"]
def test_analyze_for_multiple_faces(): def test_analyze_for_multiple_faces_in_one_image():
img = "dataset/img4.jpg" img = "dataset/img4.jpg"
# Copy and combine the same image to create multiple faces # Copy and combine the same image to create multiple faces
img = cv2.imread(img) img = cv2.imread(img)
@ -147,4 +148,13 @@ def test_analyze_for_multiple_faces():
logger.debug(demography) logger.debug(demography)
assert demography["age"] > 20 and demography["age"] < 40 assert demography["age"] > 20 and demography["age"] < 40
assert demography["dominant_gender"] == "Woman" assert demography["dominant_gender"] == "Woman"
logger.info("✅ test analyze for multiple faces done") logger.info("✅ test analyze for multiple faces in one image done")
def test_batch_detect_emotion_for_multiple_faces():
img = "dataset/img4.jpg"
img = cv2.imread(img)
imgs = [img, img]
results = DeepFace.demography.Emotion.EmotionClient().predict(imgs)
# Check two faces emotions are the same
assert np.array_equal(results[0], results[1])
logger.info("✅ test batch detect emotion for multiple faces done")