tensorflow - 解析在 tensorflow 摘要对象上评估的 `summary_str` 字节字符串

Question

目前tensorflow的tensorboard is not compatible with python3 。因此，一般来说，我正在寻找一种每 100 个时期打印一次摘要读数的方法。

是否有一个函数可以将以下几行中生成的 summary_str 字节字符串解析为 float 字典？

summary_op = tf.merge_all_summaries()
summary_str = sess.run(summary_op, feed_dict=feed_dict)

Answer 1

您可以通过将 summary_str 解析为 tf.Summary 来获取其文本表示形式。 Protocol Buffer 如下:

summary_proto = tf.Summary()
summary_proto.ParseFromString(summary_str)
print(summary_proto)

然后您可以将其转换为将字符串标签映射到 float 的字典:

summaries = {}
for val in summary_proto.value:
    # Assuming all summaries are scalars.
    summaries[val.tag] = val.simple_value