1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
|
import sys
import cv2 as cv
# ---------------------------------------------------------------------------
#
# OpenCV drawing functions
#
# ---------------------------------------------------------------------------
pose_types = {'pitch': (0,0,255), 'roll': (255,0,0), 'yaw': (0,255,0)}
def draw_landmarks2D(im, points, radius=3, color=(0,255,0), stroke_weight=2):
'''Draws facial landmarks, either 5pt or 68pt
'''
for x,y in points:
cv.circle(im, (x,y), radius, color, -1, cv.LINE_AA)
def draw_landmarks3D(im, points, radius=3, color=(0,255,0), stroke_weight=2):
'''Draws 3D facial landmarks
'''
for x,y,z in points:
cv.circle(im, (x,y), radius, color, -1, cv.LINE_AA)
def draw_bbox(im, bbox, color=(0,255,0), stroke_weight=2):
'''Draws a dimensioned (not-normalized) BBox onto cv2 image
'''
cv.rectangle(im, bbox.pt_tl, bbox.pt_br, color, stroke_weight)
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.75, color, thickness=1, lineType=cv.LINE_AA)
def draw_degrees(im, pose_data, color=(0,255,0)):
'''Draws degrees as text over image
'''
for i, pose_type in enumerate(pose_types.items()):
k, clr = pose_type
v = pose_data[k]
t = '{}: {:.2f}'.format(k, v)
origin = (10, 30 + (25 * i))
cv.putText(im, t, origin, cv.FONT_HERSHEY_SIMPLEX, 0.5, clr, thickness=2, lineType=2)
|
