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}')