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