From c67a3c23190287f747989e5dc7725e1657edf8f1 Mon Sep 17 00:00:00 2001 From: Adam Harvey Date: Mon, 7 Jan 2019 18:49:09 +0100 Subject: add age, gender, emotion stubs --- megapixels/app/processors/face_recognition.py | 9 +++++++++ 1 file changed, 9 insertions(+) (limited to 'megapixels/app/processors/face_recognition.py') diff --git a/megapixels/app/processors/face_recognition.py b/megapixels/app/processors/face_recognition.py index a56b1634..76f00aa1 100644 --- a/megapixels/app/processors/face_recognition.py +++ b/megapixels/app/processors/face_recognition.py @@ -54,6 +54,15 @@ class RecognitionDLIB: #vec = self.facerec.compute_face_descriptor(im, face_shape) return vec + def flatten(self, vec): + '''Converts 128D vector into a flattened list for CSV + :param points: (list) a feature vector as list of floats + :returns dict item for each point (eg {'d1':0.28442156, 'd1': 0.1868632}) + ''' + vec_flat = {} + for idx, val in enumerate(vec, 1): + vec_flat[f'd{idx}'] = val + return vec_flat def similarity(self, query_enc, known_enc): return np.linalg.norm(query_enc - known_enc, axis=1) -- cgit v1.2.3-70-g09d2