def layers(net):
out = []
for elem in net['layer']:
out.append(elem['name'])
for x in elem['top']:
out.append(x)
return set(out)
微信
公众号
扫码关注公众号