解析在tensorflow摘要对象上计算的`summary_str`字节字符串

时间:2016-01-02 17:31:37

标签: tensorflow

目前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)

1 个答案:

答案 0 :(得分:11)

您可以通过将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

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

def op1():
   print('1')

def op2():
   print('2')

def op3():
   print('3')

if option == 'A':
   op1()
elif option == 'B'::
   op2()
else:
   op3()