python - TensorFlow输入数据 Protocol Buffer (tf.train.Example)具有字符串类型的Feature的TypeError

我试图根据张量流的tf.train.Example将数据编码为tutorial。
我有一个要传递给Features类的Example属性的字符串值，并且正在使用以下代码：

import tensorflow as tf
tf_example = tf.train.Example()
s1 = "sample string 1"
tf_example.features.feature['str1'].bytes_list.value.extend([s1])

但是，我得到的错误是期望bytes而不是str：

TypeError: 'sample string 1' has type <class 'str'>, but expected one of: ((<class 'bytes'>,),)

我想念什么？

最佳答案

看来他们希望s1是一个字节字符串，因此您需要在b之前添加"：

import tensorflow as tf
tf_example = tf.train.Example()
s1 = b"sample string 1"
tf_example.features.feature['str1'].bytes_list.value.extend([s1])

关于python - TensorFlow输入数据 Protocol Buffer (tf.train.Example)具有字符串类型的Feature的TypeError，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/44665255/