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

[fix] model input size -> (n, w, h, c)

Browse Source
This commit is contained in:
NatLee 2025-01-07 05:26:07 +08:00
parent 431544ac52
commit 041773232f
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -73,7 +73,10 @@ class EmotionClient(Demography):
imgs = self._preprocess_batch_or_single_input(img)
# Preprocess each 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)
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
# Emotion model input shape is (48, 48, 1, n), where n is the batch size