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)