EncodeGenerator.py

import cv2
import face_recognition
import pickle
import os
import firebase_admin
from firebase_admin import credentials
from firebase_admin import db
from firebase_admin import storage

cred = credentials.Certificate("serviceAccountKey.json")
firebase_admin.initialize_app(cred , {
    'databaseURL' : "https://attendance-system-real-time-default-rtdb.firebaseio.com/",
    "storageBucket" : "attendance-system-real-time.appspot.com"
})


#importing student images
folderPath = 'images'
PathList = os.listdir((folderPath))
#print(PathList)
imgList = []
studentIds = []

for path in PathList :
    imgList.append(cv2.imread(os.path.join(folderPath , path)))
    studentIds.append(os.path.splitext(path)[0])


    fileName = f'{folderPath}/{path}'
    bucket = storage.bucket()
    blob = bucket.blob(fileName)
    blob.upload_from_filename(fileName)
    #print(path)
    #print(os.path.splitext(path)[0])
print(studentIds)


def findEncodings(imagesList) :
    encodesList = []
    for img in imagesList :
        img = cv2.cvtColor(img , cv2.COLOR_BGR2RGB)
        encode = face_recognition.face_encodings(img)[0]
        encodesList.append(encode)

    return encodesList
print("Encoding started ...")
encodeListknown = findEncodings(imgList)
encodeListknownWithIds = [encodeListknown , studentIds]
#print(encodeListknown)
print("Encoding complete")

file = open("EncodeFile.p" , 'wb')
pickle.dump(encodeListknownWithIds , file)
file.close()
print("File saved")