A.I. powered security systems could lead us to a safer world. Real-Time face detection, threat detection, face recognition, and individual detection are some main functionalities of this system.

You can find them here

• This model named

person_detection.pt

has been made for person tracking.

• This model named

threat_detection.pt

has been made for threat tracking.

• This model named

person_and_threat_detection.pt

has been made for person and threat tracking.

• This model has been made for you to give it a dataset, and then automates the training process to get the model.

In order to train it, follow these steps:

1. Give a dataset: For face recognition, we want labels/classes where the photos of individuals are.

The directory where you should put the classes images is: ./training/face_models/YOLO_data_utils/data/

    ... data/
            ---individual_1:
                photos
            ---individual_2:
                photos
            ---individual_3:
                photos
            ---individual_4:
            ...
            ...
            ...
            ...
            ...

The process is detailed on the README.md inside data folder.

2. Run main.py on face_models.

3. Wait until the model finishes the training phase.

4. Move the best.pt model from the training session you accomplished and send it to the /models/ folder.

5. Now you can access to the model and save it on /models/ directory, there is a preview executing app.py