Create generate.py

generate.py

@@ -0,0 +1,29 @@
+import os
+import face_recognition
+import pickle
+
+
+PATH_BACKGROUND = 'images_background'
+PATH_MODEL = 'bin'
+DATA_IMAGE_PICKLE = 'data_images.pkl'
+
+print('Loading data from brackground images ...')
+filename = os.path.join(os.getcwd(), PATH_MODEL, DATA_IMAGE_PICKLE)
+images_background_encoding = []
+images_background_names = []
+images_background_contents = []
+for filename_image in os.listdir(PATH_BACKGROUND):
+    if filename_image.endswith('.gitkeep'):
+        continue
+    image_path = os.path.join(PATH_BACKGROUND, filename_image)
+    image_loaded = face_recognition.load_image_file(image_path)
+    face_encoding = face_recognition.face_encodings(image_loaded)[0]
+    images_background_encoding.append(face_encoding)
+    images_background_names.append(filename_image)
+    images_background_contents.append(image_loaded)
+
+data_images = {"names": images_background_names, "encodings": images_background_encoding, 'content':images_background_contents}
+with open(filename, 'wb') as file:
+    pickle.dump(data_images, file)
+
+print(f'Generated data from background images on {filename}')