+关注
已关注

分类  

暂无分类

标签  

暂无标签

日期归档  

暂无数据

python中tf.train.Example.SerializeToString的Tensorflow Java API是什么?

发布于2021-04-12 14:48     阅读(378)     评论(0)     点赞(24)     收藏(1)


0

1

2

3

4

5

尝试将一些python张量流代码转换为java,我在python中找不到相同的java API for train.example.SerializeToString。它肯定不是java toString。我可以在这里https://www.javadoc.io/doc/org.tensorflow/proto/latest/org/tensorflow/example/package-summary.html找到Java Example类,但是根本没有诸如SerializeToString这样的API。

在Python中

features["a"] = tf.train.Feature(int64_list=tf.train.Int64List(value=list(values_a)))
features["b"] = tf.train.Feature(int64_list=tf.train.Int64List(value=list(values_b)))
tf_example = tf.train.Example(features=tf.train.Features(feature=features))
model_input = tf_example.SerializeToString()

我可以转换成Java

Feature featureA = Feature.newBuilder().setInt64List(value_a).build();
Feature featureB = Feature.newBuilder().setInt64List(value_b).build();
Map<String, Feature> featureNameAndValueMap = new HashMap<>();
featureNameAndValueMap.put("a", featureA);
featureNameAndValueMap.put("b", featureB);
Example example = Example.newBuilder().setFeatures(features).build();
//String modelInput = example.to???

非常感谢


解决方案


用这个 exampleBuilder.build().toByteString().toString("UTF-8")

0

1

2

3

4

5

6

7

8

9



所属网站分类: 技术文章 > 问答

作者:黑洞官方问答小能手

链接: https://www.pythonheidong.com/blog/article/936020/0db748611586eef38864/

来源: python黑洞网

任何形式的转载都请注明出处,如有侵权 一经发现 必将追究其法律责任

24 0
收藏该文
已收藏

评论内容:(最多支持255个字符)