Fix the spacing of labels on draw_bounding_boxes (#3895)

0bb9b914 · Vasilis Vryniotis · GitHub · 74559c47 · 74559c47 · 0bb9b914
Unverified Commit 0bb9b914 authored May 22, 2021 by Vasilis Vryniotis Committed by GitHub May 22, 2021
Show whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

test/assets/fakedata/draw_boxes_util.png test/assets/fakedata/draw_boxes_util.png +0 -0

torchvision/utils.py torchvision/utils.py +2 -1

No files found.
--- a/test/assets/fakedata/draw_boxes_util.png
+++ b/test/assets/fakedata/draw_boxes_util.png
--- a/torchvision/utils.py
+++ b/torchvision/utils.py
@@ -211,7 +211,8 @@ def draw_bounding_boxes(
            draw.rectangle(bbox, width=width, outline=color)

        if labels is not None:
-            draw.text((bbox[0], bbox[1]), labels[i], fill=color, font=txt_font)
+            margin = width + 1
+            draw.text((bbox[0] + margin, bbox[1] + margin), labels[i], fill=color, font=txt_font)

    return torch.from_numpy(np.array(img_to_draw)).permute(2, 0, 1).to(dtype=torch.uint8)