summaryrefslogtreecommitdiff
path: root/megapixels/commands/cv/face_landmark_2d_68.py
diff options
context:
space:
mode:
authorJules Laplace <julescarbon@gmail.com>2019-01-17 15:11:47 +0100
committerJules Laplace <julescarbon@gmail.com>2019-01-17 15:11:47 +0100
commit85ae432fb6c6c17292b319bca068e46a4ea81eb3 (patch)
tree4d0270fac0fdc7c1c1333af9c4bb82c6eb00669d /megapixels/commands/cv/face_landmark_2d_68.py
parentc293006ba43944ffeb4dcab17b2256f3a5491a36 (diff)
parent03ad11fb2a3dcd425d50167b15d72d4e0ef536a2 (diff)
Merge branch 'master' of github.com:adamhrv/megapixels_dev
Diffstat (limited to 'megapixels/commands/cv/face_landmark_2d_68.py')
-rw-r--r--megapixels/commands/cv/face_landmark_2d_68.py4
1 files changed, 2 insertions, 2 deletions
diff --git a/megapixels/commands/cv/face_landmark_2d_68.py b/megapixels/commands/cv/face_landmark_2d_68.py
index e24d4b60..c6978a40 100644
--- a/megapixels/commands/cv/face_landmark_2d_68.py
+++ b/megapixels/commands/cv/face_landmark_2d_68.py
@@ -126,7 +126,7 @@ def cli(ctx, opt_fp_in, opt_fp_out, opt_dir_media, opt_data_store, opt_dataset,
bbox = BBox.from_xywh(x, y, w, h).to_dim(dim)
points = landmark_detector.landmarks(im_resized, bbox)
points_norm = landmark_detector.normalize(points, dim)
- points_flat = landmark_detector.flatten(points_norm)
+ points_str = landmark_detector.to_str(points_norm)
# display if optioned
if opt_display:
@@ -137,7 +137,7 @@ def cli(ctx, opt_fp_in, opt_fp_out, opt_dir_media, opt_data_store, opt_dataset,
display_utils.handle_keyboard()
# add to results for CSV
- results.append(points_flat)
+ results.append({'vec': points_str, 'roi_index':roi_index})
# create DataFrame and save to CSV