最近学习Object Detection API的使用教程,发现很多坑,找了很多博客,现在把自己使用的过程分享给大家。
一共分为如下几个步骤:(前提是大家已经装完python、tensorflow、anaconda等环境)
1、准备模型;下载地址:https://github.com/tensorflow/models
下载后保存在本地,C:\Object Detection API\models-master(根据具体情况自己切换)
2、安装依赖环境:在cmd或者Anaconda prompt里面输入如下指令进行安装。安装前,可以通过pip list查看相关模块是否已安装
pip install pillow
pip install lxml
pip install jupyter
pip install matplotlib
3、下载protoc。
下载的官方地址:https://github.com/protocolbuffers/protobuf/releases
这个我一直下载不下来,所以自己在CSDN搜索下载的。大家可以自行搜索
下载解压后的情况如下:包含了bin、inclue文件夹,bin文件夹下面有protoc.exe文件
4、及配置环境变量:
将bin 下面的proto.exe放在windows/system32 下面
将protoc下bin文件夹的路径、步骤1下载模型下的research、research文件加下的slim,都加载到环境变量path中
5、安装protoc。
(1)首先需要安装Visual C++ 2015构建工具,地址https://download.microsoft.com/download/5/f/7/5f7acaeb-8363-451f-9425-68a90f98b238/visualcppbuildtools_full.exe,全程默认安装
(2)之后运行以下命令:
pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
出现如下字符表示安装成功
6、在anaconda prompt进入到下载模型的model/research路径下面
cd: */model/research
然后输入如下指令:
bash
protoc object_detection/protos/*.proto --python_out=.
编译成功后,models\research\object_detection\protos目录下会出现很多.py文件
然后输入如下指令,安装object_detection模块
pip install .
7、在anaconda prompt进入到下载模型的model/research/object_detection/colab_tutorials路径下面,然后输入::
jupyter notebook
打开object_detection_tutorial.ipynb
8、前面的install过程都可以略过,直接从Imports部分开始运行
这时候就可以正常运行了。需要更换模型的话,可以参考Model preparation里面的指引,在detection model zoo获取支持的模型列表,可以通过更换model_name,实验不同模型的效果
以上就是我的一些使用经验,希望帮到大家