解析在张量流摘要对象上评估的“summary_str”字节字符串

Question

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

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

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

Answer 1

您可以将 summary_str 的文本表示形式解析为 tf.Summary 协议缓冲区，如下所示：

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

然后您可以将其转换为将字符串标签映射到浮点数的字典：

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

Parsing `summary_str` byte string evaluated on tensorflow summary object