diff options
Diffstat (limited to 'megapixels/commands/demo/face_vector.py')
| -rw-r--r-- | megapixels/commands/demo/face_vector.py | 7 |
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) |