Add projections and average face to the model

1 files changed, 7 insertions, 1 deletions

diff --git a/train.py b/train.py
index a0c2a9a..6c17cc7 100755
--- a/train.py
+++ b/train.py
@@ -33,4 +33,10 @@ normal_faces = normalise_faces(average_face, raw_faces)
 
 e_vals, e_vecs = LA.eig(np.cov(normal_faces))
 
-np.savez(args.model, e_vals=e_vals[:M], e_vecs=e_vecs[:M])
+np.savez(args.model, 
+         average_face=average_face, 
+         e_vals=e_vals[:M], 
+         e_vecs=e_vecs[:M], 
+         projections=np.dot(e_vecs[:M], raw_faces)
+        )
+