From 1afdc762b80613f9ae8c6f35296a883c40fdc375 Mon Sep 17 00:00:00 2001 From: Vasil Zlatanov Date: Thu, 18 Oct 2018 12:54:21 +0100 Subject: Require -m argument --- train.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'train.py') diff --git a/train.py b/train.py index b7e9fb0..a0c2a9a 100755 --- a/train.py +++ b/train.py @@ -18,13 +18,13 @@ def normalise_faces(average_face, raw_faces): parser = argparse.ArgumentParser() parser.add_argument("-i", "--data", help="Input CSV file", required=True) parser.add_argument("-o", "--model", help="Output model file", required=True) -parser.add_argument("-m", "--M", help="Number of eigenvalues in model", type=int) +parser.add_argument("-m", "--eigen", help="Number of eigenvalues in model", required=True, type=int) args = parser.parse_args() assert args.data, "No input CSV data (-i, --input-data)" assert args.model, "No model specified (-o, --model)" -M = args.M | -1; +M = args.eigen raw_faces = genfromtxt(args.data, delimiter=',').T -- cgit v1.2.3-54-g00ecf