1 files changed, 9 insertions, 0 deletions
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)