基于TensorFlow Object Detection API使用ssd_mobilenet_v1训练自己的数据集并转换为tflite模型（亲测可用）

1. 环境

OS: Ubuntu 16.04 x64
Anaconda: 4.6.12
Python: 3.6.8
TensorFlow（GPU版）: 1.13.1
OpenCV: 3.4.1

2. 基础环境配置

Anaconda 下载地址: Anaconda-4.6.12-Linux

本文中安装位置为 /usr/local/anaconda3

修改默认的 python 版本为 3.6

conda install python=3.6

安装 OpenCV 3.4.1

conda install opencv=3.4.1

倘若安装 TensorFlow 1.13.1（GPU版），首先需要安装合适的NVIDIA的驱动，重启电脑以后执行以下命令可以看到需要下载对应版本的cudatoolkit、cudnn、tensorflow-gpu等依赖包，使用conda安装不需要我们自己去配环境，输入y就可以直接安装。

conda install tensorflow-gpu==1.13.1

倘若安装 TensorFlow 1.13.1（CPU版），直接执行以下命令。

conda install tensorflow=1.13.1

3. TensorFlow Models

下载地址: Github - TensorFlow Models
下载后得到一个 models-master.zip 文件，解压后移动到 /usr/local/anaconda3/lib/python3.6/site-packages/tensorflow文件夹下，并重命名为 models

unzip models-master.zip
mv models /usr/local/anaconda3/lib/python3.6/site-packages/tensorflow

进入 models/research目录，并编译 protobuf

cd /usr/local/anaconda3/lib/python3.6/site-packages/tensorflow/models/research
protoc object_detection/protos/*.proto --python_out=.

安装 object_detection 库

python setup.py build
python setup.py install

设置 PYTHONPATH

export PYTHONPATH=$PYTHONPATH:/usr/local/anaconda3/lib/python3.6/site-packages/tensorflow/models/research
export PYTHONPATH=$PYTHONPATH:/usr/local/anaconda3/lib