From c67a3c23190287f747989e5dc7725e1657edf8f1 Mon Sep 17 00:00:00 2001 From: Adam Harvey Date: Mon, 7 Jan 2019 18:49:09 +0100 Subject: add age, gender, emotion stubs --- megapixels/app/utils/draw_utils.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) (limited to 'megapixels/app/utils/draw_utils.py') diff --git a/megapixels/app/utils/draw_utils.py b/megapixels/app/utils/draw_utils.py index 47bb7978..cafac5a5 100644 --- a/megapixels/app/utils/draw_utils.py +++ b/megapixels/app/utils/draw_utils.py @@ -34,12 +34,18 @@ def draw_bbox(im, bbox, color=(0,255,0), stroke_weight=2): def draw_pose(im, pt_nose, image_pts): '''Draws 3-axis pose over image - ''' + ''' cv.line(im, pt_nose, tuple(image_pts['pitch'].ravel()), pose_types['pitch'], 3) cv.line(im, pt_nose, tuple(image_pts['yaw'].ravel()), pose_types['yaw'], 3) cv.line(im, pt_nose, tuple(image_pts['roll'].ravel()), pose_types['roll'], 3) +def draw_text(im, pt, text, color=(0,255,0)): + '''Draws degrees as text over image + ''' + cv.putText(im, text, pt, cv.FONT_HERSHEY_SIMPLEX, 0.5, color, thickness=2, lineType=2) + + def draw_degrees(im, pose_data, color=(0,255,0)): '''Draws degrees as text over image ''' -- cgit v1.2.3-70-g09d2