TensorFlow Object Detection API 安装使用

谷歌开源的 TensorFlow Object Detection API 玩起来还不错,虽然我还没深度的剖析其内部的东西,这里只是做个记录,防止自己以后忘记,重新捡起来,浪费时间~

当时,我把自带的demo跑起来的时候,可是花了不少的时间,而且遇到了不少的坑~

一步一步重新捡起来吧~

一 、下载模型

直接给出链接:https://github.com/tensorflow/models
(版本一直在更新~)
有装Git的话,就直接clone下来,么有的话就下载下来吧,会有点慢~
内容长这样~
在这里插入图片描述

二 、安装protoc

为什么要安装这个呢?
那是因为我们想把proto文件编译为python文件,(注意:protoc的版本大于2.6.0)
获取地址:https://github.com/protocolbuffers/protobuf/releases
这里面有所有版本~
我这里选择的是3.7.0的~
最好版本是不要太新啦~
下载后解压得到:
在这里插入图片描述
我把它放在了C:\Program Files\protoc-3.7.0-win64这个路径下面~
然后配置环境变量~

在这里插入图片描述
在这里插入图片描述
配好了之后,控制台输入protoc --version查看版本好~
在这里插入图片描述
ok,这就装好了,那就开始转换文件吧~

在目录文件夹下面 \models\research\执行:

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

得到python文件:
在这里插入图片描述

三、Anaconda下配置

我是参考这篇文章的:
连接:https://blog.csdn.net/qq_28019591/article/details/82023949

在Anaconda(或者你tensorflow安装的环境)\Lib\site-packages添加一个路径文件,如tensorflow_model.pth,必须以.pth为后缀,写上你要加入的模块文件所在的目录名称,如下图:
在这里插入图片描述
这里的路径是你model/research存放的路径,自行修改

运行models/research下的setup.py

python setup.py build

python setup.py install

在models/research下运行如下命令:

python object_detection/builders/model_builder_test.py

出现如下信息,说明已安装成功:
在这里插入图片描述

四、执行已经训练好的模型

在research文件夹下运行命令:jupyter notebook,接着在jupyter中打开object_detection文件夹,并单击object_detection_tutorial.ipynb运行试例文件。如下所示:
在这里插入图片描述

运行所有代码后的结果如下:
在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

龚大龙

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值