官方推出了Tensorflow Lite的轉(zhuǎn)換工具tflite_convert
傀广,但是這個(gè)工具無(wú)法用于轉(zhuǎn)換ssd_mobilenet_v1的文件,出現(xiàn)難以解決的問(wèn)題翘单,如果小伙伴知道吨枉,可以也告訴我蹦渣,所以還是用bazel
工具進(jìn)行轉(zhuǎn)換。本文是在我上一篇文章的后續(xù)文章东羹,可以參考下剂桥。
http://www.reibang.com/p/031239631bf7
環(huán)境
tensorflow = 1.12.0
bazel = 0.18.1
ubuntu = 18.04.1
python = 3.6.2
安裝 bazel (0.18.1)
如果tensorflow是1.12.0
,那么必須安裝指定版本0.18.1
的bazel属提,不然會(huì)出現(xiàn)很多的錯(cuò)誤無(wú)法解決权逗。
wget https://github.com/bazelbuild/bazel/releases/download/0.18.1/bazel-0.18.1-installer-linux-x86_64.sh
chmod +x bazel-0.18.1-installer-linux-x86_64.sh
./bazel-0.18.1-installer-linux-x86_64.sh --user
更多安裝方式參考 https://docs.bazel.build/versions/master/install.html
下載tensorflow工程代碼
git clone https://github.com/tensorflow/tensorflow.git
編譯轉(zhuǎn)換工具
cd tensorflow/
bazel build tensorflow/python/tools:freeze_graph
bazel build tensorflow/contrib/lite/toco:toco
如果操作系統(tǒng)和選擇的bazel不同,這個(gè)環(huán)節(jié)會(huì)出現(xiàn)各種錯(cuò)誤冤议,所以選擇合適的軟件版本特別重要斟薇。
生成tflite_graph.pb文件
cd models/research/object_detection
python export_tflite_ssd_graph.py \
--pipeline_config_path=data/ssd_mobilenet_v1_coco.config \
--trained_checkpoint_prefix=data/training/model.ckpt-28189 \
--output_directory=data/output \
--add_postprocessing_op=true
利用bazel生成tflite文件
bazel-bin/tensorflow/contrib/lite/toco/toco \
--input_file=tflite_graph.pb \
--output_file=detect.tflite \
--input_shapes=1,300,300,3 \
--input_arrays=normalized_input_image_tensor \
--output_arrays='TFLite_Detection_PostProcess','TFLite_Detection_PostProcess:1','TFLite_Detection_PostProcess:2','TFLite_Detection_PostProcess:3' \
--inference_type=QUANTIZED_UINT8 \
--mean_values=128 \
--std_values=128 \
--change_concat_input_ranges=false \
--default_ranges_min=0 \
--default_ranges_max=6 \
--allow_custom_ops
在Android上測(cè)試
在下面的目錄中有tensorflow lite的例子,可以替換原來(lái)的detect.tflite
文件恕酸,修改對(duì)應(yīng)的coco_labels_list.txt
文件堪滨,建議改成不一樣的名稱,修改代碼蕊温,不然運(yùn)行的時(shí)候袱箱,detect.tflite
和coco_labels_list.txt
會(huì)重新下載,又被覆蓋掉了义矛。
https://github.com/tensorflow/tensorflow/tree/master/tensorflow/lite/examples/android