added helper function for showing text on frame

Signed-off-by: Mpho Mphego <mpho112@gmail.com>

Author: mmphego

main.py

@@ -176,6 +176,7 @@ def main(args):
                 predict_end_time, head_pose_angles = head_pose_estimation.predict(
                     face, show_bbox=True
                 )
+                head_pose_estimation.show_text(frame, head_pose_angles)
                 text = f"Head Pose Est. Inference time: {predict_end_time:.3f} s"
                 head_pose_estimation.add_text(
                     text, frame, (15, video_feed.source_height - 40)

src/model.py

@@ -377,6 +377,25 @@ def draw_output(coords, image):
 
         return image
 
+    @staticmethod
+    def show_text(
+        image, coords, pos=500, font_scale=1.5, color=(255, 255, 255), thickness=1
+    ):
+        """Helper function for showing the text on frame."""
+        height, _ = image.shape[:2]
+        ypos = abs(height - pos)
+        text = ", ".join(f"{x}: {y:.2f}" for x, y in coords.items())
+
+        cv2.putText(
+            image,
+            text,
+            (15, ypos),
+            fontFace=cv2.FONT_HERSHEY_PLAIN,
+            fontScale=font_scale,
+            color=color,
+            thickness=thickness,
+        )
+
 
 class Gaze_Estimation(Base):
     """Class for the Gaze Estimation Detection Model."""