Keras:已经知道图层名称时如何获取图层索引?

Ngu*_*inh 3 python keras

我已经知道模型层的名称,现在我想知道该层的索引。有没有可用的功能可以做到这一点?谢谢你们。

Akh*_*esh 5

假设您的模型是modellayerName层的名称是。

index = None
for idx, layer in enumerate(model.layers):
    if layer.name == layerName:
        index = idx
        break
Run Code Online (Sandbox Code Playgroud)

index是必填名称的IDX。


man*_*iac 5

Akhilesh 作为函数的答案:

def getLayerIndexByName(model, layername):
    for idx, layer in enumerate(model.layers):
        if layer.name == layername:
            return idx
Run Code Online (Sandbox Code Playgroud)