Tensorflow Object detection Api入门指南

虽然已经有这么多的资料来介绍如何使用Tensorflow的物体检测接口,但是一般都是工程性质的,没有对相关的细节进行研究。我自然也没有研究很多的细节,但是对其中感兴趣的部分进行了些许的研究。所以就想自己来写一下整个操作过程中的心得体会,一来是对自己工作的一个总结,二来或许会帮助一些正在填坑的童鞋。

首先给出一些其他的介绍物体检测Api的资源:

安装及训练流程

Raccoon detector
TensorFlow —— 30秒搞定物体检测
Deep Dive into Object Detection with Open Images, using Tensorflow
Mac安装tensorflow Object detection Api
使用tensorflow object detection api训练自己的数据

Train a TensorFlow Object Detection Classifier for multiple object detection on Windows
这是window上的版本,作者还在Youtube上上传了教学视频,手把手教大家如何操作。

GPU&CPU 并行加速

ssd mobilenet v1 在TX2上加速,此链接给出的是通过将训练好的ssd模型中的卷积部分和NMS部分分别在GPU和CPU上运行来加速模型的inference速度的,实测至少可以达到18FPS. 相关的讨论参见here

Tensorflow Lite:
Training and serving a realtime mobile object detector in 30 minutes with Cloud TPUs
这是2018.7.14由谷歌官方博客给出的详细的利用tensorflow object detection api & TPU训练自己数据,并且将模型转换为tf-lite模型移植到手机端的很详细的手把手教程。其实也不一定非要用TPU,我自己在mac上测试过,关键是从源代码安装 TensorFlow容易出问题,大家可以小心的试一下。

Tensorflow Mobile
当然你也可以不用Tf-Lite, 而是直接将电脑端训练好的模型应用在手机端,可以参考A GUIDE TO RETRAINING OBJECT DETECTION MODELS WITH TENSORFLOW或者参考官网的
Building TensorFlow on Android

计划在博客中以此为题展开一个系列的讲解,包括
Tensorflow Api的安装
标注数据获取及格式转换
模型训练流程
ssd-mobilenet v1 算法结构及代码介绍

  • 1
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
TensorFlow Object Detection API 是一个开源项目,它提供了一系列基于 TensorFlow 的工具和库,用于实现目标检测任务。对于 macOS 系统,我们可以通过以下步骤来使用 TensorFlow Object Detection API: 1. 安装 TensorFlow:在 macOS 上安装 TensorFlow 是使用 TensorFlow Object Detection API 的前提。你可以通过 pip 命令进行安装,例如在终端中执行 `pip install tensorflow`。 2. 下载 TensorFlow Object Detection API:打开终端并导航到适合你的工作目录中,然后使用 git 命令来克隆 TensorFlow Object Detection API 的 GitHub 仓库,例如执行 `git clone https://github.com/tensorflow/models.git`。 3. 安装依赖项:进入克隆的模型目录中,找到 research 文件夹并进入。然后运行 `pip install -r object_detection/requirements.txt` 命令来安装所需的依赖项。 4. 下载预训练模型:在 TensorFlow Object Detection API 中,我们可以使用预训练的模型来进行目标检测。你可以从 TensorFlow Model Zoo 中下载适合你任务的模型,并将其解压到你的工作目录中。 5. 运行实例代码:在 research/object_detection 目录中,你可以找到一些示例代码,用于训练、评估和使用目标检测模型。可以通过阅读这些示例代码并根据自己的需求进行修改。例如,你可以使用 `python object_detection/builders/model_builder_tf2_test.py` 命令来运行一个模型的测试。 以上是在 macOS 上使用 TensorFlow Object Detection API 的基本步骤,你可以根据你的具体需求进行更多的深入研究和调整。希望这些信息能帮助到你!

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值