Spaces:
Running
Running
File size: 1,881 Bytes
7b53d75 | 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 58 59 60 61 62 63 64 65 | import cv2
import numpy as np
from .results import GazeResultContainer
def draw_gaze(a,b,c,d,image_in, pitchyaw, thickness=2, color=(255, 255, 0),sclae=2.0):
"""Draw gaze angle on given image with a given eye positions."""
image_out = image_in
(h, w) = image_in.shape[:2]
length = c
pos = (int(a+c / 2.0), int(b+d / 2.0))
if len(image_out.shape) == 2 or image_out.shape[2] == 1:
image_out = cv2.cvtColor(image_out, cv2.COLOR_GRAY2BGR)
dx = -length * np.sin(pitchyaw[0]) * np.cos(pitchyaw[1])
dy = -length * np.sin(pitchyaw[1])
cv2.arrowedLine(image_out, tuple(np.round(pos).astype(np.int32)),
tuple(np.round([pos[0] + dx, pos[1] + dy]).astype(int)), color,
thickness, cv2.LINE_AA, tipLength=0.18)
return image_out
def draw_bbox(frame: np.ndarray, bbox: np.ndarray):
x_min=int(bbox[0])
if x_min < 0:
x_min = 0
y_min=int(bbox[1])
if y_min < 0:
y_min = 0
x_max=int(bbox[2])
y_max=int(bbox[3])
cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), (0,255,0), 1)
return frame
def render(frame: np.ndarray, results: GazeResultContainer):
# Draw bounding boxes
for bbox in results.bboxes:
frame = draw_bbox(frame, bbox)
# Draw Gaze
for i in range(results.pitch.shape[0]):
bbox = results.bboxes[i]
pitch = results.pitch[i]
yaw = results.yaw[i]
# Extract safe min and max of x,y
x_min=int(bbox[0])
if x_min < 0:
x_min = 0
y_min=int(bbox[1])
if y_min < 0:
y_min = 0
x_max=int(bbox[2])
y_max=int(bbox[3])
# Compute sizes
bbox_width = x_max - x_min
bbox_height = y_max - y_min
draw_gaze(x_min,y_min,bbox_width, bbox_height,frame,(pitch,yaw),color=(0,0,255))
return frame
|