main.py

"""
Author: Rodolfo Ferro @ Datlacuache
Twitter: @rodo_ferro

Script: Main script.
"""

from tools.dataset import extract_frames_from_folder
from tools.facial import extract_faces_from_image
from tools.dataset import extract_frames_from_video

def main():
    """Main function."""

    # Input directory
    input_dir = 'assets'

    # Output directory
    output_dir = 'frames'

    # Extract frames
    extract_frames_from_folder(
        input_dir,
        output_dir,
        prefix_name='img',
        starting_id=1,
        frame_step=30,  # Used for fps
        extension='png')


if __name__ == '__main__':
    # Run main function
    # main()

    # Extract faces
    extract_faces_from_image('assets/frames/img_0000011867.jpg',
                             'haarcascades/haarcascade_frontalface_alt.xml',
                             'faces')

    # Extract frames from video
    # extract_frames_from_video('assets/sesion.mp4',
    #                          'assets/frames')