一、生成.pb文件
如果后续要在移动端部署,将模型转换成tflite,那么训练完的模型导出时需要调用文件object_detection下的export_tflite_ssd_graph.py:
python export_tflite_ssd_graph.py --input_type image_tensor --pipeline_co
nfig_path training/ssdlite_mobilenet_v3_small_320x320_coco.config --trained_checkpoint_prefix training/model.ckpt-80000 --output_directory zuanjing_tflite_inference_graph
二、配置 bazel
安装bazel
首先下载的是3.1.0版本的bazel(3.5.0版本的试了出错不知道为什么?),打开https://github.com/bazelbuild/bazel/releases/download/3.1.0/bazel-3.1.0-installer-linux-x86_64.sh,下载好后cd到bazel-3.1.0-installer-linux-x86_64.sh所在的目录下,依次运行
chmod +x bazel-3.1.0-installer-linux-x86_64.sh
./bazel-3.1.0-installer-linux-x86_64.sh --user
sudo gedit ~/.bashrc
# 在文件最后面添加路径
export PATH="$PATH:$HOME/bin"
source ~/.bashrc
测试bazel