python - TF 对象检测 API - 删除标签和百分比；仅边界框

Question

我已经训练了自己的模型，它可以检测自定义对象。但我需要一些调整。

如何删除标签和百分比？ 示例图像如下所示。我只想显示边界框。

Answer 1

第 190 行:在 /util/visualization.py 内:

for display_str in display_str_list[::-1]:
    text_width, text_height = font.getsize(display_str)
    margin = np.ceil(0.05 * text_height)
    print("\n"+display_str+"\n")
    #draw.rectangle(
        #[(left, text_bottom - text_height - 2 * margin), (left + text_width,
                                                     # text_bottom)],
        #fill=color)
    #draw.text(
    # (left + margin, text_bottom - text_height - margin),
    # display_str,
    # fill='black',
    # font=font)
    text_bottom -= text_height - 2 * margin

您可以注释绘制文本及其周围矩形的线条。