1

有没有办法按照nolearn 提供的方式打印 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()))

我需要完成表格,以某种方式爬行或走上“层堆栈”(即从给定层到传入/输入层)。

4

1 回答 1

0

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

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

于 2017-02-05T22:21:22.110 回答