tensorflow - 自定义 TF Object Detection 边界框厚度和标签字体大小

Question

使用 https://github.com/tensorflow/models/blob/master/research/object_detection/object_detection_tutorial.ipynb 运行预测.

• 操作系统平台和发行版(例如，Linux Ubuntu 16.04):

  Linux Ubuntu 16.04

想要自定义标签字体大小和边界框厚度，因为我的标签文本和 bbox 在图像检测中太厚了。

感谢您的帮助!如果你自己做过这个，请传递你的学习! :)

Answer 1

更改字体大小:

在文件 models/research/object_detection/utils/visualization_utils.py 从第 202 行开始:

试试:

  font = ImageFont.truetype('arial.ttf', 24) 
except IOError:
  font = ImageFont.load_default()

这里我们只需要将数字 24 更改为所需的字体大小。