Update viz_utils.py

viz_utils.py

@@ -8,6 +8,11 @@ import numpy as np
 import pdb
 from datetime import date
 today = date.today()
+FONTS = {'amiko': "fonts/Amiko-Regular.ttf",
+        'nature': "fonts/LoveNature.otf", 
+        'painter':"fonts/PainterDecorator.otf",
+        'animals': "fonts/UncialAnimals.ttf", 
+        'zen': "fonts/ZEN.TTF"}
 
 #########################################
 #  Draw keypoints on image
@@ -16,6 +21,7 @@ def draw_keypoints_on_image(image,
                             map_label_id_to_str,
                             flag_show_str_labels,
                             use_normalized_coordinates=True,                           
+                            font_style='amiko',
                             font_size=8,
                             keypt_color="#ff0000",
                             marker_size=2,
@@ -67,15 +73,18 @@ def draw_keypoints_on_image(image,
 
         # add string labels around keypoints
         if flag_show_str_labels:
+            font = ImageFont.truetype(FONTS[font_style], 
+                                     font_size) 
             draw.text((keypoint_x + marker_size, keypoint_y + marker_size),#(0.5*im_width, 0.5*im_height), #-------
                       map_label_id_to_str[i],
-                      ImageColor.getcolor(keypt_color, "RGB") # rgb #
-                      )
+                      ImageColor.getcolor(keypt_color, "RGB"), # rgb #
+                      font=font)
 
 #########################################
 #  Draw bboxes on image
 def draw_bbox_w_text(img,
                      results,
+                     font_style='amiko',
                      font_size=8): #TODO: select color too?
     #pdb.set_trace()
     bbxyxy = results
@@ -86,6 +95,7 @@ def draw_bbox_w_text(img,
 
     confidence = bbxyxy[4]
     string_bb = 'animal ' + str(round(confidence, 2))
+    font = ImageFont.truetype(FONTS[font_style], font_size) 
 
     text_size = font.getsize(string_bb) # (h,w)
     position = (bbxyxy[0],bbxyxy[1] - text_size[1] -2 )