有没有办法按照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()))
我需要完成的表格,将是一些爬行或走上层叠的方法" (即从给定层到达传入/输入层。
答案 0 :(得分:0)
您寻求的不是文字表示,但也许TensorBoard就足够了?您可以使用此工具可视化整个计算图并监控模型。
https://www.tensorflow.org/how_tos/summaries_and_tensorboard/