tensorflow - Tensorboard直方图转换为matplotlib

Question

我想“转储”张量板直方图并通过matplotlib绘制它们。我会有更多的科学论文吸引人的情节。

我设法使用tf.train.summary_iterator破解了摘要文件，并转储了我想转储的直方图(tensorflow.core.framework.summary_pb2.HistogramProto对象)。
通过这样做并实现Java脚本代码对数据(https://github.com/tensorflow/tensorboard/blob/c2fe054231fe77f3a5b05dbc519f713d2e738d1c/tensorboard/plugins/histogram/tf_histogram_dashboard/histogramCore.ts#L104)的作用，我设法通过张量板图获得了类似的结果(相同趋势)，但并没有获得完全相同的图。

我可以对此有所了解吗？

谢谢

Answer 1

为了用matplotlib绘制张量板直方图，我正在做以下事情:

event_acc = EventAccumulator(path, size_guidance={
    'histograms': STEP_COUNT,
})
event_acc.Reload()
tags = event_acc.Tags()
result = {}
for hist in tags['histograms']:
    histograms = event_acc.Histograms(hist)
    result[hist] = np.array([np.repeat(np.array(h.histogram_value.bucket_limit), np.array(h.histogram_value.bucket).astype(np.int)) for h in histograms])
return result

h.histogram_value.bucket_limit给我值， h.histogram_value.bucket给我值。因此，当我相应地重复这些值( np.repeat(...))时，我得到了一个巨大的预期大小数组。现在可以使用默认的matplotlib逻辑绘制该数组。