summaryrefslogtreecommitdiff
path: root/megapixels/commands/demo/face_vector.py
diff options
context:
space:
mode:
authorAdam Harvey <adam@ahprojects.com>2019-01-07 18:49:09 +0100
committerAdam Harvey <adam@ahprojects.com>2019-01-07 18:49:09 +0100
commitc67a3c23190287f747989e5dc7725e1657edf8f1 (patch)
tree5ff47ab72e6fb146f66730db06fc3b91265dd769 /megapixels/commands/demo/face_vector.py
parent55b9734d131a197166156566d1b999a8bb59169b (diff)
add age, gender, emotion stubs
Diffstat (limited to 'megapixels/commands/demo/face_vector.py')
-rw-r--r--megapixels/commands/demo/face_vector.py7
1 files changed, 4 insertions, 3 deletions
diff --git a/megapixels/commands/demo/face_vector.py b/megapixels/commands/demo/face_vector.py
index 1104f923..3ff68001 100644
--- a/megapixels/commands/demo/face_vector.py
+++ b/megapixels/commands/demo/face_vector.py
@@ -68,10 +68,11 @@ def cli(ctx, opt_fp_in, opt_gpu, opt_size, opt_display):
# generate face vectors, only to test if feature extraction works
from app.processors import face_recognition
- face_rec = face_recognition.RecognitionDLIB()
- vec = face_rec.vec(im_resized, bbox_dim)
+ facerec = face_recognition.RecognitionDLIB()
+ vec = facerec.vec(im_resized, bbox_dim)
+ vec_flat = facerec.flatten(vec)
log.info(f'generated vector. showing vec[0:10]:')
- log.info(f'\n{vec[0:10]}')
+ log.info(f'\n{vec_flat}')
if opt_display:
draw_utils.draw_bbox(im_resized, bbox_dim)