打印Tensorflow(tf-slim)模型的文本表示

时间:2016-09-07 00:04:12

标签: tensorflow tf-slim

有没有办法按照nolearn offers的方式打印tf-slim模型的文本表示:

## Layer information

name            size          total    cap.Y    cap.X    cov.Y    cov.X    filter Y    filter X    field Y    field X
--------------  ----------  -------  -------  -------  -------  -------  ----------  ----------  ---------  ---------
input           1x144x192     27648   100.00   100.00   100.00   100.00         144         192        144        192
Conv2DLayer     12x144x192   331776   100.00   100.00     2.08     1.56           3           3          3          3
Conv2DLayer     12x144x192   331776    60.00    60.00     3.47     2.60           3           3          5          5
MaxPool2DLayer  12x72x96      82944    60.00    60.00     3.47     2.60           3           3          5          5
...
DenseLayer      7                 7   100.00   100.00   100.00   100.00         144         192        144        192

编辑:

我可以使用这样的东西来打印给定图层的信息:

print("%s: %s" % (layer.name, layer.get_shape()))

我需要完成的表格,将是一些爬行或走上层叠的方法" (即从给定层到达传入/输入层。

1 个答案:

答案 0 :(得分:0)

您寻求的不是文字表示,但也许TensorBoard就足够了?您可以使用此工具可视化整个计算图并监控模型。

https://www.tensorflow.org/how_tos/summaries_and_tensorboard/