gpt4 book ai didi

tensorflow - 在 go 中使用 tensorflow hub

转载 作者:数据小太阳 更新时间:2023-10-29 03:19:03 25 4
gpt4 key购买 nike

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

但是我找不到任何关于如何将 tensorflow hub 与 tensorflow 的 go 语言绑定(bind)结合使用的文档。

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

最佳答案

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

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

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

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

关于tensorflow - 在 go 中使用 tensorflow hub,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/56787254/

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com