Font arial

run_opencv.py

@@ -35,7 +35,7 @@ class ArgParser(Tap):
 
     id2label: Optional[str] = "labels/kinetics_400.json"
 
-    threshold: Optional[float] = 10.0
+    threshold: Optional[float] = None # 10.0
 
     max_confidence: Optional[float] = 20.0  # Set None if not scale
 

utils/frame_rate.py

@@ -3,6 +3,7 @@ from typing import Optional
 
 import cv2
 import numpy as np
+from PIL import Image, ImageDraw, ImageFont
 
 
 class FrameRate:
@@ -12,6 +13,7 @@ class FrameRate:
         self.NO_FRAMES = 10
         self.fps: float = -1
         self.label: str = ""
+        self.font = ImageFont.truetype("fonts/arial.ttf", 33)
         self.reset()
 
     def reset(self) -> None:
@@ -29,15 +31,21 @@ class FrameRate:
 
     def show_fps(self, image: np.ndarray, is_recording=False) -> np.ndarray:
         if self.fps != -1:
-            image = cv2.putText(
-                image,
-                f"FPS {self.fps:.0f} _ {self.label}",
-                (50, 50),
-                cv2.FONT_HERSHEY_SIMPLEX,
-                fontScale=1,
-                color=(255, 0, 0),
-                thickness=2,
-            )
+            text = f"FPS {self.fps:.0f} _ {self.label}"
+            # image = cv2.putText(
+            #     image,
+            #     text,
+            #     (50, 50),
+            #     cv2.FONT_HERSHEY_SIMPLEX,
+            #     fontScale=1,
+            #     color=(255, 0, 0),
+            #     thickness=2,
+            # )
+            pil_image = Image.fromarray(image)
+            draw = ImageDraw.Draw(pil_image)
+            draw.text((50, 50), text, font=self.font)
+            image = np.asarray(pil_image)
+
             if is_recording:
                 image = cv2.circle(
                     image, (50, 100), radius=10, color=(0, 0, 255), thickness=-1