First test

Dockerfile

@@ -0,0 +1,11 @@
+FROM python:3.9
+
+WORKDIR /code
+
+COPY ./requirements.txt /code/requirements.txt
+
+RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt
+
+COPY . .
+
+CMD ["gunicorn", "-b", "0.0.0.0:7860", "main:app"]

main.py

@@ -0,0 +1,56 @@
+from flask import Flask, request, jsonify
+import os
+import face_recognition
+
+app = Flask(__name__)
+
+@app.route('/trouver_personnes', methods=['POST'])
+def trouver_personnes():
+    # Vérifier si le fichier d'image est présent dans la requête
+    if 'image' not in request.files:
+        return jsonify({'message': 'Aucun fichier d\'image trouvé.'}), 400
+
+    image_a_comparer = request.files['image']
+
+    # Charger l'image à comparer
+    image = face_recognition.load_image_file(image_a_comparer)
+    encodage_image = face_recognition.face_encodings(image)
+
+    if len(encodage_image) == 0:
+        return jsonify({'message': 'Aucun visage détecté dans l\'image.'}), 400
+
+    encodage_image = encodage_image[0]
+
+    correspondances = []
+
+    # Parcourir les images dans le dossier identite_photo
+    for fichier in os.listdir('BD_photo'):
+        if fichier.endswith('.jpg') or fichier.endswith('.png'):
+            chemin_image = os.path.join('BD_photo', fichier)
+            image_base = face_recognition.load_image_file(chemin_image)
+            encodage_base = face_recognition.face_encodings(image_base)
+
+            if len(encodage_base) == 0:
+                continue
+
+            encodage_base = encodage_base[0]
+
+            # Comparer les encodages faciaux
+            correspondance = face_recognition.compare_faces([encodage_base], encodage_image)
+
+            if correspondance[0]:
+                nom_personne = os.path.splitext(fichier)[0]
+                taux_fiabilite = face_recognition.face_distance([encodage_base], encodage_image)[0]
+                # correspondances.append((nom_personne, taux_fiabilite))
+                correspondances.append(nom_personne)
+
+    # Trier les correspondances par ordre décroissant de taux de fiabilité
+    correspondances = sorted(correspondances, key=lambda x: x[1], reverse=True)
+
+    if len(correspondances) > 0:
+        return jsonify({'resultats': correspondances})
+    else:
+        return jsonify({'message': 'Aucune correspondance trouvée.'}), 404
+
+if __name__ == '__main__':
+    app.run()

requirements.txt

@@ -0,0 +1,15 @@
+blinker==1.6.2
+click==8.1.7
+cmake==3.21.4
+colorama==0.4.6
+dlib==19.24.2
+face-recognition==1.3.0
+face-recognition-models==0.3.0
+Flask==2.3.3
+itsdangerous==2.1.2
+Jinja2==3.1.2
+MarkupSafe==2.1.3
+numpy==1.25.2
+Pillow==10.0.0
+Werkzeug==2.3.7
+gunicorn