[BugFix] Update pillow usage (#16269)

[BugFix] Update textsize usage
apache · Jan 2, 2024 · c157ca1 · c157ca1
1 parent 8b2cff8
commit c157ca1
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/python/tvm/relay/testing/yolo_detection.py b/python/tvm/relay/testing/yolo_detection.py
@@ -315,10 +315,12 @@ def _get_label(font_path, labelstr, rgb):
     from PIL import ImageFont
 
     text = labelstr
+    textSize = 25
     colorText = "black"
     testDraw = ImageDraw.Draw(Image.new("RGB", (1, 1)))
-    font = ImageFont.truetype(font_path, 25)
-    width, height = testDraw.textsize(labelstr, font=font)
+    font = ImageFont.truetype(font_path, textSize)
+    width = int(testDraw.textlength(labelstr, font=font))
+    height = textSize + 5
     img = Image.new(
         "RGB", (width, height), color=(int(rgb[0] * 255), int(rgb[1] * 255), int(rgb[2] * 255))
     )