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

patch: emotion dimension.

Browse Source
This commit is contained in:
h-alice 2025-01-13 23:44:28 +08:00
parent 72b6db19d6
commit a23893a5fa
No known key found for this signature in database
GPG Key ID: 5708F34144A70909
1 changed files with 1 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
deepface/models/demography/Emotion.py
View File

@ -72,14 +72,7 @@ class EmotionClient(Demography):
# Preprocessing input image or image list. # Preprocessing input image or image list.
imgs = self._preprocess_batch_or_single_input(img) imgs = self._preprocess_batch_or_single_input(img)
if imgs.shape[0] == 1: processed_imgs = np.expand_dims(np.array([self._preprocess_image(img) for img in imgs]), axis=-1)
# Preprocess single image and add channel dimension for grayscale images
processed_imgs = np.expand_dims(np.array([self._preprocess_image(img) for img in imgs]), axis=0)
else:
# Preprocess batch of images and add channel dimension for grayscale images
processed_imgs = np.expand_dims(np.array([self._preprocess_image(img) for img in imgs]), axis=-1)
# Reshape input for model (expected shape=(n, 48, 48, 1)), where n is the batch size
processed_imgs = processed_imgs.reshape(processed_imgs.shape[0], 48, 48, 1)
# Prediction # Prediction
predictions = self._predict_internal(processed_imgs) predictions = self._predict_internal(processed_imgs)