Mask-RCNN 分类计数

原创于 2023-12-17 20:59:20 发布 · 95 阅读

CC 4.0 BY-SA版权

文章标签：

文章讲述了如何在`draw_box_utils.py`中修改代码，避免在绘制文本时出现白色边界框。通过删除或注释`draw.rectangle`调用，改用计算精确的`text_width`和`margin`来调整`draw.text`的位置，从而实现无框效果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

修改文件：

draw_box_utils.py

不希望在绘制的文本周围有白色边界框

可以将以下代码块中的 draw.rectangle 函数调用注释掉或删除掉：

draw.rectangle([(left, text_top),
                (left + text_width + 2 * margin, text_bottom)], fill=None)

改成：

    for ds in display_str:
        text_width, text_height = font.getsize(ds)
        margin = np.ceil(0.05 * text_width)
        # draw.rectangle([(left, text_top),
        #                 (left + text_width + 2 * margin, text_bottom)], fill=None)
        draw.text((left + margin, text_top),
                  ds,
                  fill=color,
                  font=font)
        left += text_width