python – 使用keras训练TF模型,然后在Go中进行评估

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python – 使用keras训练TF模型,然后在Go中进行评估，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含2395字，纯文字阅读大概需要4分钟。

内容图文

我正在尝试使用keras设置经典的MNIST挑战模型,然后保存张量流图并随后将其加载到Go中并使用一些输入进行评估.我一直在关注this article,它提供了github的完整代码.Nils只使用tensorflow来设置comp.graph,但我想使用keras.我管理员以与他相同的方式保存模型

模型：

   model = Sequential()
model.add(Conv2D(32, kernel_size=(3, 3),
                 activation='relu',
                 input_shape=(28,28,1), name="inputNode"))
model.add(Conv2D(64, (3, 3), activation='relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(0.25))
model.add(Flatten())
model.add(Dense(128, activation='relu'))
model.add(Dropout(0.5))
model.add(Dense(10, activation='softmax', name="inferNode"))

运行正常,训练和评估,然后保存如上所述：

builder = tf.saved_model.builder.SavedModelBuilder("mnistmodel_my")
# GOLANG note that we must tag our model so that we can retrieve it at inference-time
builder.add_meta_graph_and_variables(sess, ["serve"])
builder.save()

然后,我尝试评估为：

result, runErr := model.Session.Run(
        map[tf.Output]*tf.Tensor{
            model.Graph.Operation("inputNode").Output(0): tensor,
        },
        []tf.Output{
            model.Graph.Operation("inferNode").Output(0),
        },
        nil,
    )

在Go中我按照示例进行操作,但在评估时,我得到：

    panic: nil-Operation. If the Output was created with a Scope object, see Scope.Err() for details.

goroutine 1 [running]:
github.com/tensorflow/tensorflow/tensorflow/go.Output.c(0x0, 0x0, 0x0, 0x0)
    /Users/air/go/src/github.com/tensorflow/tensorflow/tensorflow/go/operation.go:119 +0xbb
github.com/tensorflow/tensorflow/tensorflow/go.newCRunArgs(0xc42006e210, 0xc420047ef0, 0x1, 0x1, 0x0, 0x0, 0x0, 0xc4200723c8)
    /Users/air/go/src/github.com/tensorflow/tensorflow/tensorflow/go/session.go:307 +0x22d
github.com/tensorflow/tensorflow/tensorflow/go.(*Session).Run(0xc420078060, 0xc42006e210, 0xc420047ef0, 0x1, 0x1, 0x0, 0x0, 0x0, 0x0, 0x0, ...)
    /Users/air/go/src/github.com/tensorflow/tensorflow/tensorflow/go/session.go:85 +0x153
main.main()
    /Users/air/PycharmProjects/GoTensor/custom.go:36 +0x341
exit status 2

因为它说nil-Operation我认为我可能错误地标记了节点.但我不知道我应该标记哪些其他节点？

非常感谢！！！

解决方法:

你的代码应该工作正常.你对nil-operation的原因是正确的.

您只需要找到“inputNode”的完整节点名称.

从python开始,在模型定义之后,您可以循环遍历图形节点并以这种方式查找完整名称：

for n in sess.graph.as_graph_def().node:
    if "inputNode" in n.name:
        print(n.name)

获得完整名称后,即可在Go程序中使用它.

另外,我建议您在tensorflow API：tfgo周围使用更完整且易于使用的包装器

内容总结

以上是互联网集市为您收集整理的python – 使用keras训练TF模型,然后在Go中进行评估全部内容，希望文章能够帮你解决python – 使用keras训练TF模型,然后在Go中进行评估所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端

---