登录
首页 >  Golang >  Go问答

结合使用 TensorFlow Hub 和 Go

来源:stackoverflow

时间:2024-03-08 10:27:25 396浏览 收藏

最近发现不少小伙伴都对Golang很感兴趣,所以今天继续给大家介绍Golang相关的知识,本文《结合使用 TensorFlow Hub 和 Go》主要内容涉及到等等知识点,希望能帮到你!当然如果阅读本文时存在不同想法,可以在评论中表达,但是请勿使用过激的措辞~

问题内容

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

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

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


解决方案


经过这几天的努力,我终于找到了办法。

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

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

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

今天带大家了解了的相关知识,希望对你有所帮助;关于Golang的技术知识我们会一点点深入介绍,欢迎大家关注golang学习网公众号,一起学习编程~

声明:本文转载于:stackoverflow 如有侵犯,请联系study_golang@163.com删除
相关阅读
更多>
最新阅读
更多>
课程推荐
更多>