From 280fa1877a77633084401f922985d2655fef3db7 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Sat, 22 May 2021 09:06:00 +0100 Subject: [PATCH 1/2] Fix spacing on text. --- torchvision/utils.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/torchvision/utils.py b/torchvision/utils.py index 8b23cae6eee..12787433781 100644 --- 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) From c84e5520c6d053741d1589dc0f284296070ff097 Mon Sep 17 00:00:00 2001 From: Vasilis Vryniotis Date: Sat, 22 May 2021 10:15:35 +0100 Subject: [PATCH 2/2] Update expected file. --- test/assets/fakedata/draw_boxes_util.png | Bin 547 -> 560 bytes 1 file changed, 0 insertions(+), 0 deletions(-) diff --git a/test/assets/fakedata/draw_boxes_util.png b/test/assets/fakedata/draw_boxes_util.png index 2c361c5fafd52abb6a6b645e92fb4fe972a716ac..d38f8be78ac72d8d9e3cfeb110bc141c78b645c6 100644 GIT binary patch delta 465 zcmZ3?vVmoSO8s|F7srr_IdAVAENpU+aClfRt98lhQj_`$Am0ma!@GHVpMCzjB=yF$ zEaRu&<;?H?xcKnWtm9fQ*y8J5w>~>LOXPJCzAZ!_DP5B;1Nt_ocxecbn;V1 zL1b(Ov;4q`y%}ozefrE7FZ{k}=HFeBMUqU%fR3wSVzmy;e&^O?&j19Tu6{1-oD!M< DcU;^k delta 448 zcmdnMvY2IpO8qNO7srr_IdAVA%spfv;qdUjXt4pynTw(p4=qp06`bNabJ02ch0@E$ zDRQ@NRfb(@s=U8asL9QL+vA81lVdaZG^2X;lRsC#Kc}VCcmDV`^XJ8JUy}0f2D`l3 z{`>#Fp5Gt7+!rtKUl#HE?Y)norbn&z&O7uo!uV*_mZiJSimd+GR{tYl-?cRz9_>#A zv?G=}@8VmXY_ZSljupc(rXE%aZUeCmTKngiADw$dEoe$#`LP$T7?ulZE0w+9eLqg> zUfZEOGxz*xv*n5jIt%Q0+rJk4Tx=jOe?;q)JKyY%4hwgi;~V52{c)dOI{T8vw7iJv zvtw@?C->dCdAH!*tf|W1%gYznrybpu^2uk>z5};koLbAApmQLiVIxB_Bh2~)orG&E zov+^bXM0-u+ghKeHw(T=dUqwq_fPoQt~~8mbIED@7dm&AyIx=8CmpM6zNt=X``p45 z#ynY|8QsTr=|#_77k^s!!(RP}hbAr%AEDSW`6Z*|