keras 获取指定层的输出model.get_layer(p_name).output

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接: https://blog.csdn.net/weixin_38145317/article/details/99547447
在keras中,要想获取层的输出的各种信息,可以先获取层对象,再通过层对象的属性output或者output_shape获取层输出的其他特性.
获取层对象的方法为:
def get_layer(self, name=None, index=None):

函数功能:根据层的名称(这个名称具有唯一性)或者索引号检索层.若同时提供了名字name和索引好index,则以索引号为准.

检索是水平图遍历(自下而上)的检索方式.

参数:

name:层的名称

index:整数, 层的索引号

返回值:层实例(因为每个层都是一个类,所以返回的层本质上是一个类)

具体为:

1,1 特定层输出

model.get_layer(index=0).output
model.get_layer(index=0).output_shape

1.2 所有层的输出

for i in range(len(model.layers)):
  print(model.get_layer(index=i).output)

retinanet用法举例:获取特征金字塔层的输出

model.get_layer(p_name).output
 

猜你喜欢

转载自blog.csdn.net/weixin_38145317/article/details/99547447