假设您的模型是model,layerName层的名称是。
index = None
for idx, layer in enumerate(model.layers):
if layer.name == layerName:
index = idx
break
Run Code Online (Sandbox Code Playgroud)
这index是必填名称的IDX。
Akhilesh 作为函数的答案:
def getLayerIndexByName(model, layername):
for idx, layer in enumerate(model.layers):
if layer.name == layername:
return idx
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
2917 次 |
| 最近记录: |