From f111cb002b9c6065fdf6bb274ce5857a9e875e8c Mon Sep 17 00:00:00 2001
From: chenshijun <csj_sky@126.com>
Date: 星期三, 05 六月 2019 15:38:49 +0800
Subject: [PATCH] face rectangle
---
code/demo2.py | 3 ++-
1 files changed, 2 insertions(+), 1 deletions(-)
diff --git a/code/demo2.py b/code/demo2.py
index 16c2c30..025eb1c 100644
--- a/code/demo2.py
+++ b/code/demo2.py
@@ -104,6 +104,7 @@
frame_num += 1
for frames_parameter in frames_parameters:
yaw_predicted, pitch_predicted, roll_predicted, x_max, x_min, y_max, y_min, bbox_height = frames_parameter
+ cv2.rectangle(frame, (int(x_min), int(y_min)), (int(x_max), int(y_max)), (0,255,0), 1)
utils.draw_axis(frame, yaw_predicted, pitch_predicted, roll_predicted, tdx = (x_min + x_max)/2, tdy= (y_min + y_max)/2, size = bbox_height*4)
cv2.imshow("frame", frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
@@ -162,7 +163,7 @@
frames_parameters.append((yaw_predicted, pitch_predicted, roll_predicted, x_max, x_min, y_max, y_min, bbox_height))
utils.draw_axis(frame, yaw_predicted, pitch_predicted, roll_predicted, tdx = (x_min + x_max)/2, tdy= (y_min + y_max)/2, size = bbox_height*4)
# Plot expanded bounding box
- # cv2.rectangle(frame, (x_min, y_min), (x_max, y_max), (0,255,0), 1)
+ cv2.rectangle(frame, (int(x_min), int(y_min)), (int(x_max), int(y_max)), (0,255,0), 1)
cv2.imshow("frame", frame)
if cv2.waitKey(1) & 0xFF == ord('q'):
break
--
Gitblit v1.8.0