@产品大学生活程序员

谷歌目标检测与识别API在macOS下快速实现

2018-06-10  本文已影响40人  EurekaTesla

导读:深度学习做图像识别有很多不同的途径。谷歌最发布Tensorflow的物体识别API(Object Detection API),让计算机视觉在各方面都更进了一步,我们可以利用这API来训练自己的数据集实现特殊的目标检测。但对于深度学习的初学者,在学习理论知识知识已经够枯燥乏味了,更不用说动手去实现实际效果并解决其中配置的问题。本教程是在macOS系统下实现的,后期由于研究需要,会在Windows系统下配置,到时候会出Windows系统下的教程,请大家及时关注。

        小编目前是准研究生,研究方向是基于深度学习的计算机视觉,对于大家在配置的过程中出现的问题,请及时在评论区留言,我会及时与大家交流。小编本着谦虚学习和对学术上严谨的态度在这里和大家交流,若有不足之处,请大家及时指出。


■ 安装环境: 


一、安装Python 、TensorFlow和其他依赖项


pip install tensorflow

pip install pillow

pip install lxml

pip install jupyter

pip install matplotlib



二、安装 Protoc, 进入Protoc下载页,下载对应的编译好的zip包。

        下载后bin目录下会有一个protoc二进制文件,覆盖到对应目录:


cp Documents/深度学习/Tensorflow/protoc-3.5.1-osx-x86_64/bin/protoc/usr/local/bin/protoc


注意:cp的文件是自己存放下载Protoc目录下的bin/protoc,根据自己存放下载文件的目录修改即可,并且拷贝到/usr/local/bin(可以读写)



三、从github上下载目标检测API的源代码

选择目录下源代码(笔者选择的目录:Documents/深度学习/Tensorflow),下载文件较大(几百兆),请耐心等待


cd Documents/深度学习/Tensorflow

git clonehttps://github.com/tensorflow/models.git




四、编译Protobuf,进入目录models/research,运行命令进行编译:


cd models/research

protoc object_detection/protos/*.proto --python_out=.




五、在当前目录下,添加slim环境变量


export PYTHONPATH=$PYTHONPATH:`pwd`:`pwd`/slim




六、测试目标检测API是否安装成功


python object_detection/builders/model_builder_test.py


在终端出现类似如下结果即安装成功 :



七、测试模型

1.启动:

此处需要说明:如果你没有安装jupyter开发环境请按照如下步骤安装(1)(2),若已经安装直接跳到步骤(3)。

(1)Homebrew 安装,复制命令到终端,开始自动下载并安装,过程中需要输入密码,其他无需任何操作:


/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"



(2)在终端输入命令


brew install jupyter   


(3)进入目录models/research/objection_detection(具体路径和自己前面下载保存的路径一致


cd /Users/nikolatesla/Documents/深度学习/Tensorflow/models/research/object_detection

jupyter notebook


(4)点击:object_detection_tutorial进入

(5)点击:Cell 下的 Run All, 耐心等待几分钟 

测试结果: 

八、修改文件路径,即可以测试自己的图片

训练的照片在models/research/object_detection/test_images下

(1)将需要训练的照片放在该目录下,并按照“imageN.jpg”命名,从1开始编号,直到第N张。

(2)在 jupyter 里面修改代码,只需将划线部分换成N+1 



九、修改输出图像的大小(以英寸为单位) 

有问题及时在评论区留言

也欢迎大家加我微信和微博相互交流和学习

上一篇下一篇

猜你喜欢

热点阅读