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/datasets.py |    3 ++-
 1 files changed, 2 insertions(+), 1 deletions(-)

diff --git a/code/datasets.py b/code/datasets.py
index 8d05d98..e8ab9f4 100644
--- a/code/datasets.py
+++ b/code/datasets.py
@@ -184,7 +184,8 @@
         img = img.crop((int(x_min), int(y_min), int(x_max), int(y_max)))
 
         # We get the pose in radians
-        pose = utils.get_ypr_fro    # Head pose from AFLW2000 datasetp.pi
+        pose = utils.get_ypr_from_mat(mat_path)
+        pitch = pose[0] * 180 / np.pi
         yaw = pose[1] * 180 / np.pi
         roll = pose[2] * 180 / np.pi
 

--
Gitblit v1.8.0