Keras：已知图层名称时如何获取图层索引？

Question

我已经知道模型图层的名称，现在我想知道该图层的索引。 有没有可用的功能可以做到这一点？ 谢谢大家。

Answer 1

假设您的模型是

model

，

layerName

是图层的名称。

index = None
for idx, layer in enumerate(model.layers):
    if layer.name == layerName:
        index = idx
        break

这里

index

是所需名称的idx。

Answer 2

Akhilesh 作为函数的答案：

def getLayerIndexByName(model, layername):
    for idx, layer in enumerate(model.layers):
        if layer.name == layername:
            return idx

Answer 3

一个简单的方法如下：

layer_names = [layer.name for layer in model.layers]
layer_idx = layer_names.index(your_layer_name)

如果你在keras中进行可视化，也可以通过如下方式实现：

from vis.utils import utils
layer_idx = utils.find_layer_idx(model, your_layer_name)

Answer 4

单行解决方案：

model.layers.index(model.get_layer(layer_name))