安装Tensorflow Object Detection API对我这种小白来说,真实太费劲了,在CSDN看了N多人的方法以后,每个人的方法都有点不一样,但是发现都不行?真的难受,然后开始开始自己摸索,最后总算是安装成功了(花了整整一天时间,心疼自己)。最后发现自己的方法步骤是看过的最少的(和GitHub上安装指导差不多)。
安装环境
既然要安装Tensorflow Object Detection API,相信都已经安装好了Anaconda+TensorFlow。如果没有的话,建议参考这篇博客:https://blog.csdn.net/Eppley/article/details/79297503。
我的环境是最新的Anaconda和Python3.7.2。Windows10系统。
下载资源
1.tensorflow/models
不建议直接到GitHub上下载(不会翻墙的下载非常慢),这里分享我的网盘资源:
链接:https://pan.baidu.com/s/1v7j07_KXY2tOLAIJcXK7YQ
提取码:uhpc
下载后解压,对于Object Detection API,只有research文件夹是有用的(为了简洁其它的都可以删除,当然大佬们可以留着深入学习)。把research文件移动到你想要的位置(建议放在接近根目录的地方,方面后续的操作)。
2.protoc
经过本人实践,建议不要下载最新的版本。我下载的是protoc-3.4.0-win3。下载地址:https://github.com/google/protobuf/releases/tag/v3.4.0
下载后解压,然后把bin文件夹下面的protoc.exe复制到C:\Windows\System32目录下(需要权限)。
开始安装
1.编译proto文件
打开Anaconda Prompt,激活tensorflow,进入到research文件夹下,输入:
protoc object_detection/protos/*.proto --python_out=.
如图所示,表示编译成功。这时可以到research/object_detection/protos下查看,每个proto文件都会对应一个py文件。
2.添加环境变量
把/research和/research/slim添加到环境变量,名称为PYTHONPATH.
3.测试
在tensorflow环境下,research目录下,输入:
python object_detection/builders/model_builder_test.py
如果输出和上图相似,说明已经安装成功了(关键看红色是否出现OK,其它的不同没有关系)。如果没有出现这个结果,说明前面出错,上面的步骤针对我的电脑和环境是可以的。
如果出现error,比如缺失module,找不到module等等,可以在CSDN上找一下,基本上都有对应的解决办法。
4.运行API
接着上面的步骤,如果测试成功。则输入
jupyter notebook
需要注意的是,在这之前需要在tensorflow环境下安装jupyter。方法是:【1】打开软件Anaconda Navigator【2】选择tensorflow环境【3】install Jupyter
输入jupyter notebook后会直接进入浏览器。选择object_detection
再选择
然后点击cell->run all
然后就是等待了(需要先下载模型,大概80M左右,然后再进行识别)。
最后应该恶意得到下面这样的结果(激动人心):
Ps:如果没有报错,但是图片无法显示,请参考下面的博客(亲测有效):
https://blog.csdn.net/weixin_43345331/article/details/87924551