在输入的RGB图像上绘制检测结果。接受一个numpy数组(cv2)或PIL图像。
def plot(
self,
conf=True,
line_width=None,
font_size=None,
font=“Arial.ttf”,
pil=False,
img=None,
im_gpu=None,
kpt_radius=5,
kpt_line=True,
labels=True,
boxes=True,
masks=True,
probs=True,
show=False,
save=False,
filename=None,
):
参数:
conf (bool): 是否绘制检测置信度分数。
line_width (float, 可选): 边界框的线宽。如果为None,则按比例缩放到图像大小。
font_size (float, 可选): 文本的字体大小。如果为None,则按比例缩放到图像大小。
font (str): 用于文本的字体。
pil (bool): 是否返回PIL图像。
img (numpy.ndarray): 绘制到另一张图像。如果不是,则绘制到原始图像。
im_gpu (torch.Tensor): 形状为(1, 3, 640, 640)的GPU上归一化的图像,用于更快的蒙版绘制。
kpt_radius (int, 可选): 绘制关键点的半径。默认为5。
kpt_line (bool): 是否绘制连接关键点的线条。
labels (bool): 是否绘制边界框的标签。
boxes (bool): 是否绘制边界框。
masks (bool): 是否绘制蒙版
yolov8,接受一个numpy数组(cv2)或PIL图像,在输入的RGB图像上绘制检测结果,
于 2024-04-08 10:30:42 首次发布