tensorflow开源Tensorflow Object Detection API安装运行测试

本文详细介绍了如何安装和测试Tensorflow对象检测API,包括依赖项、安装步骤和测试过程。首先,需要安装Tensorflow 1.2.1、Protobuf、Pillow、lxml等库。然后,从GitHub下载并编译Protobuf,更新PYTHONPATH。最后,通过运行model_builder_test.py来验证安装是否成功,并提供了修改SSD Mobilenet模型进行视频检测的测试代码。
摘要由CSDN通过智能技术生成

TensorFlow对象检测API是一个建立在TensorFlow之上的开源框架,可以轻松构建,训练和部署对象检测模型。
安装:
Tensorflow对象检测API依赖于以下项:
Protobuf 2.6
Pillow 1.0
lxml
tf Slim (which is included in the “tensorflow/models” checkout)
Jupyter notebook
Matplotlib
Tensorflow
以下是安装步骤:
我是pip安装的tensorflow1.2版本,1以下版本好像不兼容该API,命令如下:
sudo pip install –upgrade
https://storage.googleapis.com/tensorflow/linux/cpu/tensorflow-1.2.1-cp27-cp27mu-manylinux1_x86_64.whl

pip安装的tensorflow里面不知道怎么没有model文件夹,这儿Tensorflow Object Detection API的model我是在guthub上下载下来https://github.com/tensorflow/models ,并解压在home目录(不用放在tensorflow目录),以下操作大多在解压后的model目录下操作

其余的库可以通过apt-get安装:
sudo apt-get install protobuf-compiler python-pil python-lxml
sudo pip install jupyter
sudo pip install matplotlib
以上命令也可以使用以下四条pip命令代替:
sudo pip install pillow
sudo pip install lxml
sudo pip install jupyter
sudo pip install matplotlib
注:安装jupyter时可能遇到错误,更新一下pip再安装,sudo -H pip install –upgrade pip

Tensorflow Object Detection API使用Protobufs来配置模型和训练参数。在使用框架之前,必须编译Protobuf库。这应该通过从下载解压的models/目录运行以下命令来完成:
protoc object_detection/protos/*.proto –python_out=.

当在本地运行时,models /和slim目录应该附加到PYTHONPATH。这可以通过从models /运行以下来完成:
export PYTHONPATH=$PYTHONPATH:pwd:pwd/slim
注意:此命令需要从您启动的每个新终端运行。如果您想避免手动运行,可以将其作为新行添加到〜/ .bashrc文件的末尾。

至此安装完毕,可以通过运行以下命令来测试是否正确安装了Tensorflow Object Detection API:
python object_detection / builders / model_builder_test.py

这儿我测试了SSDmobilenet和SSDinception等模型,提前下载好模型,修改里面

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值