Update viz_utils.py

viz_utils.py

@@ -8,11 +8,6 @@ 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
@@ -21,7 +16,6 @@ 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,
@@ -73,18 +67,15 @@ 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 #
-                      font=font)
+                      ImageColor.getcolor(keypt_color, "RGB") # rgb #
+                      )
 
 #########################################
 #  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
@@ -95,7 +86,6 @@ 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 )