re:Invent全球大会引领风向重塑未来 了解详情
写点什么

谷歌宣布 TensorFlow Object Detection API 支持 TensorFlow 2

  • 2020 年 7 月 29 日
  • 本文字数:1161 字

    阅读完需:约 4 分钟

谷歌宣布TensorFlow Object Detection API支持TensorFlow 2

谷歌宣布 TensorFlow Object Detection API(OD)支持 TensorFlow 2(TF2)。该版本包含与 eager 模式兼容的二进制文件、两个新的网络架构和针对所有受支持模型的预训练权重。


软件工程师 Vivek Rathod 和研究科学家 Jonathan Huang 在 TensorFlow 博客上对新版本的新特性做了概述。大部分工作是让现有的预训练模型与 TF2 兼容,对模型代码进行移植,使用了 Keras 层,并提供权重作为 TF2 检查点。OD 框架还支持同步分布式训练,以及用于训练、评估和导出的 eager 模式二进制文件。虽然所有的新模型和新的开发工作都只与 TF2 有关,但 TF1 仍然受支持。大多数代码模块都兼容 TensorFlow 的两个版本,不兼容的模块则分成两个版本。Rathod 和 Huang 写道:


我们的想法是让 TF2 和 Keras 的所有好处都暴露出来,同时继续为 TF1 的广大用户群提供支持。


TensorFlow Object Detection API 是“一个建立在 TensorFlow 之上的开源框架,让对象检测模型的构建、训练和部署变得更加容易”。这个框架包含了一组用于管理图像数据输入和对象检测模型接口的工具。一般来说,目标检测模型的输入是一个图像,然后输出一组方框,表示在图像中检测到的物体位置。OD API 还提供了预训练“模型动物园”,这对于开发自定义对象检测应用程序非常有用。“模型动物园”包含了几种流行的深度学习计算机视觉架构,如 MobileNetResNet,它们是基于 Common Objects in Context(COCO) 2017 数据集进行训练的。


TensorFlow 2 于 2019 年 9 月发布,采用 Keras 作为定义模型的官方高级 API。Keras 最初被设想为定义神经网络的接口,支持后端,包括 TensorFlow 和 Theano。TF2 还将 eager 执行设置为默认模式,让开发和调试变得更加容易。新的 OD API 利用了这些特性。预训练的模型使用 Keras 层重新实现,权重以 TF2 检查点格式进行保存。OD 框架中的辅助代码已经经过编译,开发人员可以通过交互式的方式来调试模型。新的 OD API 还支持同步分布式训练,可以加快大型模型的训练,同时不会降低准确性(相比异步分布式训练)。


除了移植现有模型,新版本还包含了两个新的模型架构: CenterNetEfficientDet。CenterNet 使用单点来表示对象位置,而不是方框,并具备“最佳的速度和精度权衡”(基于 COCO 数据集)。EfficientDet 是一个新的对象检测模型,比之前的 SOTA 模型体积小了 4 倍到 9 倍,使用更少的 FLOP(13 倍到 42 倍)。


一名 Reddit 用户评论道:


EfficientDet 看起来真的很有前途,它们致力于通过 TF2 让训练 OD 模型变得更容易。不过我觉得 TF 团队绝对可以投入一些资源来构建一个更简单的 Object Detection API,可以从 Torch Hub、Huggingface 甚至 Tensorflow Hub 获得灵感。


Object Detection API源代码和预训练模型可以在 GitHub 上找到。


原文链接


Google Announces TensorFlow 2 Support in Object Detection API


2020 年 7 月 29 日 11:341220

评论

发布
暂无评论
发现更多内容

计算机工业的生态链(一)

姬翔

9月日更

架构实战营模块 7 作业指导

华仔

架构实战营

北鲲云超算平台赋能蛋白设计助推生物制药行业发展

北鲲云

谁在制造“完美男性”?

脑极体

消息系统的演进:从MOM、ESB到下一代云原生的分布式消息系统

金蝶天燕云

分布式消息

justswap市值管理机器人系统软件开发技术(案例搭建)

量化系统19942438797

交易所 做市机器人 justswap

微信开源PhxQueue:高可用、高可靠、高性能的分布式队列

OpenIM

各编程语言里对 Iterator 进行修改时的对比

BlockQuant

Java Python rust Go 语言

活动推荐 | 云原生社区 Meetup 第七期深圳站开始报名啦!

CODING DevOps

Kubernetes DevOps 微服务 活动 Meetup

APM领域国产化先锋!博睿数据与麒麟、统信、中科方德完成兼容性认证

博睿数据

Python代码阅读(第25篇):将多行字符串拆分成列表

Felix

编程 Code Programing 阅读代码 -python

老板:把系统从单体架构升级到集群架构!

程序员 架构 分布式 后端 计算机

阿里P8架构师推荐一波Java程序员必读的架构书籍清单

程序员知识圈

Java 程序员 架构 面试 编程语言

京东T8Java架构师总结整理的15w字的Java面试手册,涵盖了诸多后端技术栈的面试题和答案

程序员知识圈

Java 程序员 架构 面试 编程语言

PhxSQL设计与实现(详细版)

OpenIM

恰逢金九银十!阿里P8连夜赶稿一份基于实例驱动的设计模式笔记

Java 编程 架构 面试 阿里

广告机主板定制方案能实现双屏异显或四屏异显吗?

双赞工控

安卓主板 主板定制 广告机主板

我们是如何在研发过程中控制质量的?产品质量正变得越来越重要

爱数技术范儿

大数据 软件工程

腾讯云签约广州知识城商用密码项目,助力黄建设密码产业示范区

腾讯安全云鼎实验室

腾讯云 商用密码

微信亿级用户异常检测框架的设计与实践

OpenIM

五行兼备:联想TruScale服务的太极之道

脑极体

关系型数据库,NoSQL数据库,NewSQL数据库权威整理

hanaper

多租户是一种技术

金蝶天燕云

多租户

熬了3天2夜,啃完阿里(珠峰版)Java面试笔记,直接斩获12家大厂offer,

Java架构师迁哥

你们想知道的一切,都在这里了。

ApacheDoris

Apache 开源社区

HashMap为什么是线程不安全的?

Java技术精选

用数据搭建反馈系统

石云升

数据分析 9月日更

Redis 6.0 多线程重磅发布!来了解一下吧

Linux服务器开发

数据库 redis 网络编程 Linux服务器开发 单线程

白瞟党福音!Alibaba内部最新Java开发手册(嵩山版)灵魂17问

Java 编程 架构 面试 架构师

JAVA加餐

IT蜗壳-Tango

9月日更

博睿数据云主机性能评测新增6家云厂商,8月报告亚马逊云科技登榜首

博睿数据

百亿小文件多盘缓存的设计与实现

百亿小文件多盘缓存的设计与实现

谷歌宣布TensorFlow Object Detection API支持TensorFlow 2-InfoQ