Variables | |
parser = argparse.ArgumentParser(description='set input arguments') | |
action | |
dest | |
type | |
str | |
default | |
int | |
bool | |
args = parser.parse_args() | |
string | output_fld = '' |
output_model_file | |
weight_file_path = str(args.input_model_file) | |
net_model = load_model(weight_file_path) | |
num_output = args.num_outputs | |
list | pred = [None]*num_output |
list | pred_node_names = [None]*num_output |
name | |
sess = K.get_session() | |
f = args.output_graphdef_file | |
as_text | |
list | transforms = ["quantize_weights", "quantize_nodes"] |
transformed_graph_def = TransformGraph(sess.graph.as_graph_def(), [], pred_node_names, transforms) | |
constant_graph = graph_util.convert_variables_to_constants(sess, transformed_graph_def, pred_node_names) | |
keras_to_tensorflow.action |
Definition at line 58 of file keras_to_tensorflow.py.
keras_to_tensorflow.args = parser.parse_args() |
Definition at line 79 of file keras_to_tensorflow.py.
keras_to_tensorflow.as_text |
Definition at line 163 of file keras_to_tensorflow.py.
keras_to_tensorflow.bool |
Definition at line 71 of file keras_to_tensorflow.py.
keras_to_tensorflow.constant_graph = graph_util.convert_variables_to_constants(sess, transformed_graph_def, pred_node_names) |
Definition at line 177 of file keras_to_tensorflow.py.
keras_to_tensorflow.default |
Definition at line 59 of file keras_to_tensorflow.py.
keras_to_tensorflow.dest |
Definition at line 59 of file keras_to_tensorflow.py.
keras_to_tensorflow.f = args.output_graphdef_file |
Definition at line 162 of file keras_to_tensorflow.py.
keras_to_tensorflow.int |
Definition at line 69 of file keras_to_tensorflow.py.
keras_to_tensorflow.name |
Definition at line 151 of file keras_to_tensorflow.py.
keras_to_tensorflow.net_model = load_model(weight_file_path) |
Definition at line 114 of file keras_to_tensorflow.py.
keras_to_tensorflow.num_output = args.num_outputs |
Definition at line 124 of file keras_to_tensorflow.py.
keras_to_tensorflow.output_fld = '' |
Definition at line 96 of file keras_to_tensorflow.py.
keras_to_tensorflow.output_model_file |
Definition at line 98 of file keras_to_tensorflow.py.
keras_to_tensorflow.parser = argparse.ArgumentParser(description='set input arguments') |
Definition at line 57 of file keras_to_tensorflow.py.
list keras_to_tensorflow.pred = [None]*num_output |
Definition at line 125 of file keras_to_tensorflow.py.
list keras_to_tensorflow.pred_node_names = [None]*num_output |
Definition at line 126 of file keras_to_tensorflow.py.
keras_to_tensorflow.sess = K.get_session() |
Definition at line 159 of file keras_to_tensorflow.py.
keras_to_tensorflow.str |
Definition at line 59 of file keras_to_tensorflow.py.
keras_to_tensorflow.transformed_graph_def = TransformGraph(sess.graph.as_graph_def(), [], pred_node_names, transforms) |
Definition at line 176 of file keras_to_tensorflow.py.
list keras_to_tensorflow.transforms = ["quantize_weights", "quantize_nodes"] |
Definition at line 175 of file keras_to_tensorflow.py.
keras_to_tensorflow.type |
Definition at line 59 of file keras_to_tensorflow.py.
keras_to_tensorflow.weight_file_path = str(args.input_model_file) |
Definition at line 100 of file keras_to_tensorflow.py.