Tensorflow lite模型转换过

tensorflow lite是专门为移动设备准备的深度学习架构，在tensorflow基础上优化并支持手机上的硬件加速

首先下载如下网址链接

git clone https://github.com/tensorflow/models.git

• 下载官方训练好的模型
• (e.g., cd …/ssdlite_mobilenet_v2_coco_2018_05_09)
• 执行一下代码进行转换

 python3 ./models/research/object_detection/export_tflite_ssd_graph.py --pipeline_config_path /home/rui/model_dir/ssd_mobilenet_v2_coco.config --trained_checkpoint_prefix /home/rui/model_dir/model.ckpt-122712 --output_directory /home/rui/model_dir/ --add_postprocessing_op True

• 转化完毕后，使用一下代码进行转换

tflite_convert --graph_def_file=/home/rui/model_dir/tflite_graph.pb --output_file=/home/rui/model_dir/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 --allow_custom_ops

成功后将生成的模型文件放入android sset目录下即可，并改成相应名称。