TensorBoard通过检测显示真实情况

时间:2017-10-24 14:40:12

标签: object tensorflow detection tensorboard bounding-box

我正在使用TensorFlow模型repo进行对象检测。为了评估,我使用Tensorboard显示mAP和结果 - 预测的检测对象的边界框。 我想显示地面实况边界框以及预测的边界框。我怎么能做到这一点?

2 个答案:

答案 0 :(得分:3)

您似乎想要在TensorBoard中显示预测的对象检测框旁边的地面实况边界框。我假设您正在使用image dashboard。这是一个想法。

您可以将原始图像的字节传递给py_func,它允许您包装python函数并将其用作TensorFlow操作。

py_func内,您可以使用matplotlib(使用patches.Rectangle)在图片顶部渲染框: matplotlib: how to draw a rectangle on image

然后,您可以将这些字节传递给图像摘要操作。这个GitHub项目提供了一个例子:https://github.com/vahidk/EffectiveTensorflow/blob/master/README.md#prototyping-kernels-and-advanced-visualization-with-python-ops

答案 1 :(得分:1)