...
keras_model = load_model(keras_model_filename)
...
frozen_graph = freeze_graph(
    session.graph, 
    session, 
    [out.op.name for out in keras_model.outputs], 
    save_pb_dir=save_pb_dir
)