有没有办法按照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()))
我需要完成表格,以某种方式爬行或走上“层堆栈”(即从给定层到传入/输入层)。