谷歌大脑团队日前发布新的TensorFlow API接口，能让开发者将自定义的可视化插件加入TensorBoard，相较以前更具灵活性。

雷锋网 AI科技评论将最新消息编译整理如下：

我们在2015年开源TensorFlow时就包括TensorBoard了，它是一套用于检查和理解TensorFlow模型和进程的可视化工具。Tensorboard包含一小套预决定的通用可视化工具，几乎适用于所有深度学习应用，例如观察损失随时间的变化，探索高维空间集群。

然而，在缺乏可重用API的情况下，不属于TensorFlow团队的人几乎不可能在TensorBoard中增加新的可视化工具，这使得我们错过了研究社群中很多潜在的极具创造力、美观性和有用的可视化工具。

现在，我们发布了一组一致的API接口，通过这组接口，开发者能增加自定义可视化插件到TensorBoard中，大家可以将自己开发的新的、有用的可视化工具加进来。我们希望开发人员能用这个API来扩展TensorBoard，让它覆盖更广泛的用例。

我们已经更新了TensorBoard中的现有dashboard，它们可以作为样例，大家可以使用新的API接口加入插件。现在可以直接在GitHub的tensorboard/plugins目录下看到TensorBoard包含的插件，大家可以观察下生成precision-recall曲线的新插件，如下图:

这个插件展示了标准TensorBoard插件的三部分:

• 用于为之后的可视化进行数据收集的TensorFlow summary op。

• 提供自定义数据的Python后端。

• 用TypeScript和polymer建造的TensorBoard dashboard。

另外，与其他插件一样，“pr_curves”插件提供了一个demo，用户可以查看这个demo，学习如何使用这个插件，插件作者也可以利用这个demo在开发过程中生成示例数据。

为了进一步说明插件是如何工作的，我们还创建了一个简单的TensorBoard Greeter插件，在模型运行期间，这个插件收集问候语(“你好”后面有些简单的字符串)并将他们显示出来。

我们建议大家先从Greeter插件和其他已有的插件开始探索。

Beholder是Chris Anderson在读研期间开发的，是利用TensorBoard API的一个著名示例，Beholder在训练模型期间能将数据实时用视频显示，大家可以看下面的动图。

我们期待看到研究社群接下来的创新成果。如果你计划在TensorBoard库中新增插件，希望你先通过issue tracker与我们联系，表达你的想法，我们或许可以帮助和引导你。