TensorFlow輸入數據協議緩衝區（tf.train.Example）字符串類型爲Feature的TypeError

Question

我試圖根據張量流的tutorial將我的數據編碼爲tf.train.Example。 我有，我想傳遞給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])