Comparaison faciale AI à l’aide de FaceNet, comparez deux photos et voyez s’il s’agit de la même personne.
Installation
pip install face-compare
Utilisation
Utilisez compare_faces.py
pour comparer deux images de personnes pour voir s’il s’agit de la même personne.
compare_faces.py --input-one /path/to/image_one.png --input-two /path/to/image_two.png
Sort éventuellement la sortie de l’image recadrée dans un répertoire (utile pour inspecter l’entrée du modèle AI)
compare_faces.py --input-one /path/to/image_one.png --input-two /path/to/image_two.png -s /path/to/outputs/
Étapes Impliquées
- Un classificateur en cascade est utilisé pour détecter la face dans les images d’entrée.
- La boîte englobante de cette segmentation est ensuite utilisée pour recadrer les images et introduite dans le modèle AI.
- Le modèle FaceNet calcule ensuite les intégrations d’images pour les deux images recadrées.
- Enfin, la deuxième intégration est soustraite de la première, et la norme euclidienne de ce vecteur est calculée.
- Un seuil de 0,7 est utilisé pour déterminer s’il s’agit de la même personne ou non.
Problèmes connus
Problème d’exécution CPU Uniquement
Si vous essayez d’exécuter le module sans GPU approprié, vous pouvez rencontrer le message d’erreur suivant:
tensorflow.python.framework.errors_impl.InvalidArgumentError: Default MaxPoolingOp only supports NHWC on device type CPU
Pour résoudre ce problème avec l’architecture CPU Intel, vous pouvez installer le package d’optimisation Intel TensorFlow via
pip install intel-tensorflow