import argparse
import sys
sys.path.append('peoples')
from utilitys import encode_known_faces
parser = argparse.ArgumentParser()
parser.add_argument("--train", action="store_true", help="Train on input data")
parser.add_argument(
    "--validate", action="store_true", help="Validate trained model"
)
parser.add_argument(
    "--test", action="store_true", help="Test the model with an unknown image"
)
parser.add_argument(
    "-m",
    action="store",
    default="hog",
    choices=["hog", "cnn"],
    help="Which model to use for training: hog (CPU), cnn (GPU)",
)
parser.add_argument(
    "-f", action="store", help="Path to an image with an unknown face"
)
args = parser.parse_args()


def train() :
    encode_known_faces()
    
if __name__ == "__main__":
    if args.train:
        train()
    if args.validate:
        validate(model=args.m)