tensorflow keras 查找中间tensor并构建局部子图

2019-01-14
阅读 2 分钟
3.8k
此方法可以读取层的输出,对于输出多于1个tensor的,可以指定get_layer("rpn_class").output[0:2]等确定。但是对于自定义层的中间变量,就没办法获得了,因此需要使用方法二。