Spaces:
Runtime error
Runtime error
Commit
·
1601266
1
Parent(s):
279f060
Update app.py
Browse files
app.py
CHANGED
@@ -64,9 +64,7 @@ def predict(frame):
|
|
64 |
p_pred_deg = euler[:, 0].cpu()
|
65 |
y_pred_deg = euler[:, 1].cpu()
|
66 |
r_pred_deg = euler[:, 2].cpu()
|
67 |
-
|
68 |
-
img = Image.fromarray(arr)
|
69 |
-
return img.resize(size=(500, 500))
|
70 |
|
71 |
title = "6D Rotation Representation for Unconstrained Head Pose Estimation"
|
72 |
description = "Gradio demo for 6DRepNet. To use it, simply click the camera picture. Read more at the links below."
|
|
|
64 |
p_pred_deg = euler[:, 0].cpu()
|
65 |
y_pred_deg = euler[:, 1].cpu()
|
66 |
r_pred_deg = euler[:, 2].cpu()
|
67 |
+
return utils.plot_pose_cube(frame, y_pred_deg, p_pred_deg, r_pred_deg, x_min + int(.5*(x_max-x_min)), y_min + int(.5*(y_max-y_min)), size = bbox_width)
|
|
|
|
|
68 |
|
69 |
title = "6D Rotation Representation for Unconstrained Head Pose Estimation"
|
70 |
description = "Gradio demo for 6DRepNet. To use it, simply click the camera picture. Read more at the links below."
|