Ubuntu16.04 tensorflow物体检测

1. 安装tensorflow, python等，再安装以下工具

apt-get install protobuf-compiler python-pil python-lxml
pip install jupyter
pip install matplotlib

2. 安装protobuf

git clone https://github.com/protocolbuffers/protobuf.git
cd protobuf
./autogen.sh
./configure
make
make install

  配置protobuf环境变量

vim ~/.bashrc
   export PATH=$PATH:/usr/local/bin
   export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/lib
   export PKG_CONFIG_PATH=$PKG_CONFIG_PATH:/usr/local/lib/pkgconfig
source ~/.bashrc

3. 下载tensorflow models

git clone https://github.com/tensorflow/models.git
cd models/research
protoc object_detection/protos/*.proto --python_out=.
export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim
python setup.py install

  测试（成功会显示OK)

python object_detection/builders/model_builder_test.py

4. 将LabelImg标准的xml转为csv

git clone https://github.com/datitran/raccoon_dataset.git
cd raccoon_dataset
python xml_to_csv.py

5. 生成TFRecord

python generate_tfrecord.py --csv_input=data/train_labels.csv --output_path=data/train.record --image_dir=images/
python generate_tfrecord.py --csv_input=data/test_labels.csv --output_path=data/test.record --image_dir=images/