24.02.1
|
Go to the source code of this file.
Namespaces | |
tf_frozen_model_extractor | |
Variables | |
list | strings_to_remove = ["read", "/:0"] |
dictionary | permutations = { 1 : [0], 2 : [1, 0], 3 : [2, 1, 0], 4 : [3, 2, 0, 1]} |
parser = argparse.ArgumentParser('Extract TensorFlow net parameters') | |
dest | |
type | |
str | |
required | |
True | |
help | |
False | |
default | |
action | |
storeRes | |
args = parser.parse_args() | |
graph_def = tf.GraphDef() | |
input_map | |
None | |
return_elements | |
name | |
op_dict | |
producer_op_list | |
varname = op_val.name | |
t = op_val.eval() | |
tT = t.transpose(permutations[len(t.shape)]) | |