feature_extractor.py 文件源码

def get_layer(net,layer_name):
    out = []
    for elem in net['layer']:
        out.append(elem['name'])
    out_ind = []
    for i,elem in enumerate(out):
        if elem == layer_name:
            out_ind.append(i)
    return out_ind