[Cut TF model] use this to slice a keras model in half. useful for transfer learning in large models #tf #python

get_layers_above

def get_next_level(layer,model):
  def wrap_list(val):
    if type(val) is list:
      return val
    return [val] 
  r=[]
  for output_t in wrap_list(layer.output):
    r+=[x for x in model.layers if output_t.name in [y.name for y in wrap_list(x.input)]]
  return r

def get_layers_above(cutoff_layer,model):
  visited=set()
  to_visit=set([cutoff_layer])

  while to_visit:
    layer=to_visit.pop()
    to_visit.update(get_next_level(layer,model))
    visited.add(layer)
  return list(visited)