diff --git a/README.md b/README.md index a5f9fdf14ccc21c3bddb162e33c54afcc2a581ea..7a4dae5be66d23ed318b7e81295f56555e1e530f 100644 --- a/README.md +++ b/README.md @@ -1,21 +1,21 @@ -## Projet de détection d'expression faciales - -### Description : - -Le projet a pour but de construire un programme pour détecter les visages et leurs émotions associées depuis une image ou une vidéo prise en temps réelle. Dans un second temps, il pourra être éventuellement amélioré avec la traduction des visages en avatar. - -### Fonctions des différents fichiers : - -videoCapture.py : prend une vidéo en entrée, traite chaque frame avec imageProcess.py et renvoie la vidéo traitée ainsi - -imageProcess.py : contient les fonctions pour traiter les images, qui font appel à faceAnalysis.py - -faceAnalysis.py : contient les fonctions pour analyser les visages, utilise le modèle spécifié dans config.py - -config.py : on y précise le modèle utilisé - -utils.py : quelques fonctions utiles - -data : contient les dataset, ignoré par git donc vous devez mettre les données dedans vous mêmes - -models : contient les différents modèles utilisés +## Projet de détection d'expression faciales + +### Description : + +Le projet a pour but de construire un programme pour détecter les visages et leurs émotions associées depuis une image ou une vidéo prise en temps réelle. Dans un second temps, il pourra être éventuellement amélioré avec la traduction des visages en avatar. + +### Fonctions des différents fichiers : + +videoCapture.py : prend une vidéo en entrée, traite chaque frame avec imageProcess.py et renvoie la vidéo traitée ainsi + +imageProcess.py : contient les fonctions pour traiter les images, qui font appel à faceAnalysis.py + +faceAnalysis.py : contient les fonctions pour analyser les visages, utilise le modèle spécifié dans config.py + +config.py : on y précise le modèle utilisé + +utils.py : quelques fonctions utiles + +data : contient les dataset, ignoré par git donc vous devez mettre les données dedans vous mêmes + +models : contient les différents modèles utilisés diff --git a/__pycache__/faceAnalysis.cpython-38.pyc b/__pycache__/faceAnalysis.cpython-38.pyc index 852d1c95efd6217d291a11b364dd7f1de5840464..126438fe5bd4d5fc4fb2b0ed06e8f186e4ed592d 100644 Binary files a/__pycache__/faceAnalysis.cpython-38.pyc and b/__pycache__/faceAnalysis.cpython-38.pyc differ diff --git a/config.py b/config.py index 501e27405619b029250040ed52590fa2643279f2..be19af0f9ae99b37ac9a72d907d388e1cd787b42 100644 --- a/config.py +++ b/config.py @@ -1,8 +1,8 @@ -#Name of model used -modelName = 'firstModel' - -#Emotions provided by the dataset -emotions = ["Angry", "Disgust", "Fear", "Happy", "Sad", "Suprise", "Neutral"] - -#Shape of input of the model +#Name of model used +modelName = 'firstModel' + +#Emotions provided by the dataset +emotions = ["Angry", "Disgust", "Fear", "Happy", "Sad", "Suprise", "Neutral"] + +#Shape of input of the model input_shape = (48,48,1) \ No newline at end of file diff --git a/faceAnalysis.py b/faceAnalysis.py index 8b1993a1f2275c405ee6806585875ffc8481d3cc..bbea9d365fb80688a51b87f8840608fcfdfe2a5c 100644 --- a/faceAnalysis.py +++ b/faceAnalysis.py @@ -5,7 +5,7 @@ import cv2 from utils import * from config import emotions, input_shape, modelName -model = keras.models.load_model(modelName) #Load our model +model = keras.models.load_model("models/"+modelName) #Load our model print('Model used:', modelName) def detectEmotion(face): diff --git a/models/firstModel/saved_model.pb b/models/firstModel/saved_model.pb new file mode 100644 index 0000000000000000000000000000000000000000..237e02a67eb4e7ecaf1dc345770f2adfda6f8189 Binary files /dev/null and b/models/firstModel/saved_model.pb differ diff --git a/models/firstModel/variables/variables.data-00000-of-00001 b/models/firstModel/variables/variables.data-00000-of-00001 new file mode 100644 index 0000000000000000000000000000000000000000..9bc473297013db4df59b013282db0c5e0861e11f Binary files /dev/null and b/models/firstModel/variables/variables.data-00000-of-00001 differ diff --git a/models/firstModel/variables/variables.index b/models/firstModel/variables/variables.index new file mode 100644 index 0000000000000000000000000000000000000000..baaf5a7b3890fa0b601e07395ebf19f4c8a1c93b Binary files /dev/null and b/models/firstModel/variables/variables.index differ