在Go中使用Tensorflow Hub

时间:2019-06-18 14:30:20

标签: tensorflow go tensorflow-hub

我想在我的go应用程序中使用经过预先训练的模型。特别是Inception-ResNet-v2模型。 该模型似乎仅可通过tensorflow集线器(https://www.tensorflow.org/hub/)使用。

但是我找不到任何有关如何使用带有Tenorflow的go语言绑定的tensorflow hub的文档。

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

1 个答案:

答案 0 :(得分:0)

因此,经过几天的大量工作,我终于找到了一种方法。

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

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

转换文件需要大量工作,但最终我认为这是在使用Tensorflow模型的最佳方法。