使用pip安装TensorFlow Object Detection API

TensorFlow Object Detection API的安装相当麻烦,其 官方安装指导要求使用者先克隆下整个tensorflow/models仓库,然后安装Protobuf,编译出object_detection模块,再使用pip进行安装。虽然从开发者的角度看,此安装方法足以满足在各个系统平台下安装TensorFlow Object Detection API的需求,但对于使用者来说,安装这一个API可能就需要耗费大量的时间。

解决方法

【2021-07-05更新】TensorFlow在TensorFlow 2.2.0时推出了TensorFlow Models高级API的官方PyPI项目,然和遗憾的是,截止当前,TensorFlow官方仍然只在其中纳入Official分支,Community分支和Research分支(Object Detection就在其中)还是没有得到妥善安排。也就是说,我们仍然无法使用pip轻松地安装TensorFlow Object Detection API.

为跟上TensorFlow Object Detection API的官方更新,也为TensorFlow 2.x用户提供便捷安装途径,作出一些更新:

  • 更新项目至PyPI
  • 更新至支持TensorFlow 2.2.0以上版本
  • 更新至支持TensorFlow 1.15.0
  • 移除Python 2.x版本支持
  • 移除此前提供的egg包

最新项目地址:


鉴于上述问题,我们可以将object detection api打包成whl包或egg包,然后使用pip或者easy_install安装到site-packages中即可。

我将打包好的whl包与egg包分享出来,打包环境为Raspbian 9 Stretch(Python环境是3.5.3和Python2.7.13)、Win10 64位(Python环境是3.6.6和Python2.7.15)、Manjaro Linux 64位(Python环境是3.6.6和Python2.7.15)。下载地址:Python_Packages_of_TensorFlow_Object_Detection_API

1. 安装依赖

Tensorflow Object Detection API 需要如下依赖:

[使用包管理工具安装(Linux)/系统级安装(Windows)]

  • protobuf
  • python-tk (tk in Manjaro)

[使用pip安装]

  • tensorflow
  • pillow
  • lxml
  • jupyter
  • matplotlib
  • cython
  • contextlib2
2. 安装下载的whl包或者egg包

安装whl只需执行:

pip install xxx.whl

安装egg只需执行:

easy_install xxx.egg
3. 使用

使用方法如下:

from object_detection.xxx import xxx
# e.g.:
from object_detection.utils import label_map_util, visualization_utils
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值