ubuntu16.04 TensorFlow目标检测API安装

首先安装pip,然后进行下面的操作

tensorflow现在的安装已经很简单了,直接

pip install tensorflow

然后安装object detection API所需要的库:

sudo apt-get install protobuf-compiler python-pil python-lxml
pip install -i https://pypi.doubanio.com/simple/ ipython==5.3.0
sudo pip install -i https://pypi.doubanio.com/simple/ jupyter
sudo pip install matplotlib

最后,在github上下载models:

git clone https://github.com/tensorflow/models.git

下载下来后文件名是:models-master

注意:这里有点不一样的地方,网上的教程都是直接在models/object_detection去操作的,这个版本的tensorflow这里略有改动,我也是找了好久才找到的,现在的版本把object_detection放在了research文件夹下,另外一个是网上所有基于models文件夹下的操作这里对应的都是models-master,可以下载后改名为models

最后需要把slim这个文件夹加入环境变量不然运行的时候会报错:

  File "model_builder_test.py", line 21, in <module>
    from object_detection.builders import model_builder
ImportError: No module named object_detection.builders

输入下面命令:

vi ~/.bashrc

在最后加入:

export PYTHONPATH=$PYTHONPATH:/******/models-master/research:/*****/models-master/research/slim

这里的/******/models-master/research就是你下载的路径,保存退出

source ~/.bashrc

到这里就好了,运行model_builder_test.py试试,cd到research文件夹下:

python object_detection/builders/model_builder_test.py

不报错出现如下结果就OK了

...........
----------------------------------------------------------------------
Ran 11 tests in 0.047s

OK





  • 1
    点赞
  • 4
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值