参考链接: ImageDraw.rectangle(xy, fill=None, outline=None, width=1)
ImageDraw.rectangle()函数使用说明:
在指定的图片上绘制矩形,
通过xy可以指定矩形的位置的大小,
通过outline可以指定矩形边框的颜色,
参数fill可以用来指定矩形内部填充的颜色,
参数width可以指定矩形边框的宽度.
实验代码展示:
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)
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')] # 用于填充
)
image.show()
运行结果截图:
