added faces classifying

src/main.py

@@ -4,10 +4,13 @@ import os
 from utils import *
 from face_detection import *
 from face_recognition import *
+from face_classifying import *
 
 # list all files in data/raw
 files = list_all_files("./data/raw", ["jpg", "webp"])
 
+print(files)
+
 # process each files
 faces = []
 for file_path in files:
@@ -23,12 +26,23 @@ for file_path in files:
     # detect faces file
     faces_data = detect_face_in_image(img, scale=1)
 
-    print("faces_data: ", faces_data[1])
     # extract faces from file
-    faces += extract_faces(faces_data, img, filename)
+    extracted_faces = extract_faces(faces_data, img, filename)
 
-# classify each faces
-for face in faces:
-    face.save("./data/extracted")
+    faces += extracted_faces
+
+classified_faces = classify_faces(faces)
+
+# saving result by folders
+for person, faces in classified_faces.items():
+    dir_ = f"./data/results/{person}/"
+    if not os.path.exists(dir_):
+        os.makedirs(dir_)
+    
+    for face in faces:
+        face.save(dir_)
+
+
+print("classified_faces: ", classified_faces)
 
 cv2.destroyAllWindows()