美文网首页@产品大学生活程序员
谷歌目标检测与识别API在macOS下快速实现

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

作者: EurekaTesla | 来源:发表于2018-06-10 07:19 被阅读40次

    导读:深度学习做图像识别有很多不同的途径。谷歌最发布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 



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

    有问题及时在评论区留言

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

    相关文章

      网友评论

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

        本文链接:https://www.haomeiwen.com/subject/icffeftx.html