有没有办法打印tf-slim模型的文本表示,沿着什么nolearn offers的线条
## 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()))我需要做的是在“层堆栈”中爬行或遍历(即从一个给定层到传入/输入层)。
发布于 2017-02-06 06:21:22
它不是您想要的文本表示,但也许TensorBoard就足够了?您可以使用此工具可视化整个计算图并监控您的模型。
https://www.tensorflow.org/how_tos/summaries_and_tensorboard/
https://stackoverflow.com/questions/39359478
复制相似问题