pyfacerecall

Installation

pip install pyfacerecall

Generate own face dataset using camera

You can use cli built-in cli tool to generate dataset of your own face using your camera.

python -m pyfacerecall --output person --source 0

How to use?

from pyfacerecall import FaceRecognition


if __name__ == '__main__':
    model_path = "model"
    image_path = 'test.jpeg'
    # if you already have model trained then no need to provide dataset path while initializing class like below
    # face_recognition = FaceRecognition(number_of_classes=2)
    face_recognition = FaceRecognition("./dataset/training", "./dataset/testing")
    face_recognition.training()
    face_recognition.save_model(model_path)
    # get the model and do something with that
    model = face_recognition.load_saved_model(model_path)
    # get the model prediction
    k, result = face_recognition.model_prediction(image_path, model_path, need_cropping=False)
    print(f"detected class is {k} and prediction percentage is {result}")