tensorflow - 使用 tensorflow hub 和 go

Question

我想在我的 go 应用程序中使用预先训练好的模型。尤其是 Inception-ResNet-v2 模型。这个模型似乎只能通过 tensorflow hub ( https://www.tensorflow.org/hub/ ) 获得。

但是，我找不到任何文档如何将 tensorflow hub 与 tensorflow 的 go 语言绑定一起使用。

如何在 go 中下载和使用这些模型？

Answer 1

所以经过这几天的大量工作，我终于找到了一种方法。

起初，我只想使用 Python 来完成所有 TensorFlow 的工作，然后通过 rest 服务提供结果。然而事实证明，Tensorflow Hub 提供的模型数量非常少。这对我来说是个问题，因为我必须尝试不同的模型并进行比较。

因此，我转而使用来自https://github.com/tensorflow/models的模型。有几个教程如何将数据导出到 .pb 文件。然后可以使用 gocv 在 Go 中加载这些文件。

转换文件需要做很多工作，但最后我认为这是在 go 中使用 Tensorflow 模型的最佳方式。