From 65cb506ca182272e2701136097fd00c55dc6bd69 Mon Sep 17 00:00:00 2001 From: adamhrv Date: Wed, 16 Jan 2019 13:30:16 +0100 Subject: change bbox to norm, refine face extractor --- megapixels/commands/cv/face_landmark_2d_68.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) (limited to 'megapixels/commands/cv/face_landmark_2d_68.py') 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 -- cgit v1.2.3-70-g09d2