本文记录在Windows 10下安装配置 Tensorflow Object Detection API环境的过程
安装环境:
系统:Windows 10
显卡:Nvidia 1080 ti
python版本:py3.5
Ananconda:Anaconda3-4.2.0-Windows-x86_64
1、安装Tensorflow
安装带GPU的Tensorflow 1.14.0版本 :
用管理员权限运行CMD,采用pip安装:
pip install -i -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow_gpu==1.14.0
2、下载Tensorflow Object Detection API 代码
打开网址https://github.com/tensorflow/models 点击右上角Download,如果网速慢可以先用github账户fork,然后再使用码云搬运代码再下载
。下载完毕解压。
3、安装配置Protobuf
打开网址https://github.com/google/protobuf/releases,找到3.4.0版本,其他版本据说会有各种问题,找到对应的系统版本进行下载
。下载好后解压,打开文件目录如下,将bin中的protoc.exe复制到C:/windows下。
然后以管理以权限运行cmd,cd定位到刚才下载的model-master的解压目录到research目录下;
然后输入
protoc object_detection/protos/*.proto python_out=.
如果不报错则配置成功
4、设置环境变量
将解压后的research和slim两个目录添加到PYTHONPATH环境变量中
5、安装pycocotools
由于Tensorflow object detection API 中的训练评价会用到pycocotools这个工具,因此还得安装这玩意。
安装前首先确保PC已经安装了Visual Studio 2015,否则编译安装pycocotools会遇到各种酸爽的问题。
有大神用如下方法安装
pip install git+https://github.com/philferriere/cocoapi.git#subdirectory=PythonAPI
但是我用这个方法没装成功,首先下载就特别慢,下载太慢还会中途崩溃,fuck fuck fuck 很是无语。
注意不要用 pin install pycocotools直接安装,同样会出现匪夷所思的各种问题。
我的方法是,
1)、先 fork https://github.com/philferriere/cocoapi.git ,然后用码云导入github仓库。我导出后的代码地址
https://gitee.com/vincent_wd/cocoapi
下载该代码库
2)进入PythonAPI文件夹,打开setup.py,更改:
extra_compile_args=['-std=c99']
然后保存
3)在cmd中定位到setup.py文件夹,执行以下两条指令安装:
python setup.py build_ext --inplace
python setup.py build_ext install
6、测试官方DEMO
环境配置安装完毕,在CMD中定位到models-master\research\ object_detection文件夹,输入
jupyter notebook
在浏览器中显示如下
点击 object_detection_tutorial.ipynb
点击Cell 下的 Run ALL 即可运行DEMO
参考:
https://blog.youkuaiyun.com/qq_24474463/article/details/81529695
https://blog.youkuaiyun.com/dy_guox/article/details/79081499