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

Question

我正在尝试将我的数据编码为 tf.train.Example根据tensorflow的tutorial 。我有一个字符串值，想要传递给 Example 类的 Features 属性，我使用以下代码:

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'>,),)

我错过了什么？

Answer 1

他们似乎期望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])