/
freeze_and_convert_to_tflite.sh
executable file
·49 lines (32 loc) · 2.26 KB
/
freeze_and_convert_to_tflite.sh
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
#!/bin/bash
#################################################################################################################
# Bash script to help freez tensorflow GraphDef + ckpt into FrozenGraphDef, Optimize and convert to tflite model#
# $1 = path/to/.pbtx file #
# $2 = path/to/.ckpt file #
# $3 = true/false. 'true' for .pb and 'false' for .pbtx input #
# $4 = path/to/save/frozengraph #
# $5 = graph input node name #
# $6 = graph output_node_name #
# $7 = path/to/tflite/file e.g path/to/tflite_model.lite #
# $8 = input type e.g FLOAT #
# $9 = inference (output) type (FLOAT or QUANTIZED) #
# $10 = path/to/optimized_graph.pb #
# #
# Sample usage: #
# freez_graph_tf.sh /tmp/graph.pbtx /tmp/model.ckpt-0 false /tmp/frozen.pb ArgMax #
# #
# Make sure you are runing this from the tensorflow home folder after cloning the TF repository #
# #
#################################################################################################################
#FREEZE GRAPH
#bazel build tensorflow/python/tools:freeze_graph
bazel-bin/tensorflow/python/tools/freeze_graph --input_graph=$1 --input_checkpoint=$2 --input_binary=$3 --output_graph=$4 --output_node_names=$6
#VIEW SUMARY OF GRAPH
#bazel build /tensorflow/tools/graph_transforms:summarize_graph
bazel-bin/tensorflow/tools/graph_transforms/summarize_graph --in_graph=$4
#TRANSFORM/OPTIMIZE GRAPH
#bazel build tensorflow/tools/graph_transforms:transform_graph
bazel-bin/tensorflow/tools/graph_transforms/transform_graph --in_graph=$4 --out_graph=${10} --inputs=$5 --outputs=$6 --transforms='strip_unused_nodes(type=float, shape="1,227,227,3") remove_nodes(op=Identity, op=CheckNumerics) fold_constants(ignore_errors=true) round_weights(num_steps=256) obfuscate_names sort_by_execution_order fold_old_batch_norms fold_batch_norms'
#CONVERT TO TFLITE MODEL
#bazel build /tensorflow/contrib/lite/toco:toco
bazel-bin/tensorflow/contrib/lite/toco/toco --input_format=TENSORFLOW_GRAPHDEF --input_file=${10} --output_format=TFLITE --output_file=$7 --inference_type=$9 --input_type=$8 --input_arrays=$5 --output_arrays=$6 --inference_input_type=$8 --input_shapes=1,227,227,3