zjc
/
yolov5
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Update plots.py (#4407)

This commit is contained in:
Glenn Jocher 2021-08-14 12:50:27 +02:00 committed by GitHub
parent d9f23ed6d6
commit e086347377
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
utils/plots.py
View File

@ -79,7 +79,7 @@ def plot_one_box(x, im, color=(128, 128, 128), label=None, line_thickness=3):
cv2.putText(im, label, (c1[0], c1[1] - 2), 0, tl / 3, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA) cv2.putText(im, label, (c1[0], c1[1] - 2), 0, tl / 3, [225, 255, 255], thickness=tf, lineType=cv2.LINE_AA)
def plot_one_box_PIL(box, im, color=(128, 128, 128), label=None, line_thickness=None): def plot_one_box_PIL(box, im, color=(128, 128, 128), label=None, line_thickness=3):
# Plots one bounding box on image 'im' using PIL # Plots one bounding box on image 'im' using PIL
im = Image.fromarray(im) im = Image.fromarray(im)
draw = ImageDraw.Draw(im) draw = ImageDraw.Draw(im)