【2020年最后一天致敬科比】PIL库用于目标检测

参考链接: PIL.ImageFont.truetype(font=None, size=10, index=0, encoding=’’, layout_engine=None)
参考链接: ImageDraw.text(xy, text, fill=None, font=None, anchor=None, spacing=4, align=‘left’, direction=None, features=None, language=None, stroke_width=0, stroke_fill=None, embedded_color=False)
参考链接: ImageDraw.rectangle(xy, fill=None, outline=None, width=1)
参考链接: ImageDraw.textsize(text, font=None, spacing=4, direction=None, features=None, language=None, stroke_width=0)

实验代码展示:

from PIL import Image,ImageFont, ImageDraw
import numpy as np
import colorsys
image = Image.open("bryant.jpg")
class_names = \
    ["aeroplane", "bicycle", "bird", "boat", 
    "bottle", "bus", "car", "cat", "chair", 
    "cow", "diningtable", "dog", "horse", 
    "motorbike", "person", "pottedplant", 
    "sheep", "sofa", "train", "tvmonitor"]

hsv_tuples = [
    (x / len(class_names), 1., 1.)
        for x in range(len(class_names))
]  # 获得hsv格式的不同色度

colors = list(
    map(
        lambda x: colorsys.hsv_to_rgb(*x), 
        hsv_tuples
    )
)  # 获得rgb格式的不同颜色

colors = list(
    map(
        lambda x: (int(x[0] * 255), int(x[1] * 255), int(x[2] * 255)),
        colors
    )
)  # 通过hsv格式来调整不同类别对应边框的色度


font = ImageFont.truetype(
    font='simhei.ttf',
    size=np.floor(3e-2 * 1000 + 0.5).astype('int32')
)  # 获得字体


# label = '{} {:.2f}'.format('linmazi', 0.982)
label = '{} {:.2f}'.format('林麻子', 0.982)

draw = ImageDraw.Draw(image)  
label_size = draw.textsize(label, font)
print('label_size:',label_size)  # label_size: (165, 29)
label = label.encode('utf-8')
# print(label)

for i in range(8):  # 将矩形框画得更粗一些
    draw.rectangle(
        [100+i, 150+i, 300-i, 600-i],
        outline=colors[class_names.index('car')]
    )


draw.rectangle(
    [400, 350, 500, 600],
    outline=colors[class_names.index('tvmonitor')],
    width=8
)


draw.rectangle(
    [600, 350, 750, 600],
    outline=colors[class_names.index('motorbike')],
    width=8,
    fill=colors[class_names.index('diningtable')]  # 用于填充
)

# draw.text([600, 350], str(label,'UTF-8'), fill=(0, 0, 0), font=font)
draw.text([600, 50], label.decode(), fill=(255, 0, 0), font=font)

image.show()

控制台输出结果:

Windows PowerShell
版权所有 (C) Microsoft Corporation。保留所有权利。

尝试新的跨平台 PowerShell https://aka.ms/pscore6

加载个人及系统配置文件用了 1017 毫秒。
(base) PS C:\Users\chenxuqi\Desktop\News4cxq\test4cxq\news> conda activate ssd4pytorch1_2_0
(ssd4pytorch1_2_0) PS C:\Users\chenxuqi\Desktop\News4cxq\test4cxq\news>  & 'D:\Anaconda3\envs\ssd4pytorch1_2_0\python.exe' 'c:\Users\chenxuqi\.vscode\extensions\ms-python.python-2020.12.424452561\pythonFiles\lib\python\debugpy\launcher' '53058' '--' 'c:\Users\chenxuqi\Desktop\News4cxq\test4cxq\news\test15.py'
label_size: (165, 29)
(ssd4pytorch1_2_0) PS C:\Users\chenxuqi\Desktop\News4cxq\test4cxq\news>

运行结果截图:

程序原图:

源代码及相关文件下载地址链接: 【2020年最后一天致敬科比】PIL库用于目标检测.zip