Tensorflow Object Detection API使用总结
写在前面:
因为我使用的是别人的服务器,使用API之前已经搭建好一些基本的配置,比如CUDA,CUDNN以及Tensorflow的GPU版,所以这些安装就没有放出来,简要介绍一下后面的过程。
第一步:自己找一个目录下载模型文件,两种方法
1.终端下输入git clone https://github.com/tensorflow/models
2.windows下进入https://github.com/tensorflow/models,点击clone,下载完成后将models-master文件放到服务器目录下即可。
第二步:Protobuf配置
首先要安装protoc,直接使用下面的代码一次在终端下输入即可
mkdir protoc_3.3
cd protoc_3.3
wget wget https://github.com/google/protobuf/releases/download/v3.3.0/protoc-3.3.0-linux-x86_64.zip
chmod 775 protoc-3.3.0-linux-x86_64.zip
unzip protoc-3.3.0-linux-x86_64.zip
这样就安装成功了,然后进入到自己下载模型文件的research目录下,在终端输入下面的指令
/your_path/protoc_3.3/bin/protoc object_detection/protos/*.proto --python_out=.
不报错的话可以在protos目录下看到生成的.py文件
第三步:测试是否安装成功
首先将slim路径添加到环境变量:
export PYTHONPATH=$PYTHONPATH:pwd
:pwd
/slim
然后输入以下指令进行测试:
python object_detection/builders/model_builder_test.py
出现以下结果安装成功。