diff options
Diffstat (limited to 'train.py')
-rwxr-xr-x | train.py | 8 |
1 files changed, 7 insertions, 1 deletions
@@ -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) + ) + |