当前位置：首页 > 编程日记 > 正文

港中大、商汤开源目标检测工具包mmdetection，对比Detectron如何？

编程日记 2025-01-21 16:30:01

近日，香港中文大学-商汤联合实验室开源了基于 PyTorch 的检测库——mmdetection。上个月，商汤和港中大组成的团队在 COCO 比赛的物体检测（Detection）项目中夺得冠军，而 mmdetection 正是基于 COCO 比赛时的 codebase 重构。

商汤称，这个开源库提供了已公开发表的多种视觉检测核心模块。通过这些模块的组合，可以迅速搭建出各种著名的检测框架，比如 Faster RCNN，Mask RCNN，和 R-FCN 等，以及各种新型框架，从而大大加快检测技术研究的效率。

目前，mmdetection 的第一个版本已经实现了 RPN，Fast R-CNN，Faster R-CNN，Mask R-CNN，近期还计划放出 RetinaNet 和 Cascade R-CNN。

项目地址：
https://github.com/open-mmlab/mmdetection

此次项目的参与者，香港中文大学陈恺博士称，相比 FAIR 此前开源的 Detectron，mmdetection 有以下几大优势：

编者注：Detectron 是 FAIR 用于实现最先进的目标检测算法（包括 Mask R-CNN）的软件系统。该系统基于深度学习框架 Caffe 2 ，由 Python 编写而成。（《Mask R-CNN 源代码终上线，Facebook 开源目标检测平台—Detectron》）

Performance 稍高：由于 PyTorch 官方 model zoo 里面的 ResNet 结构和 Detectron 所用的 ResNet 有细微差别（mmdetection 中可以通过 backbone 的 style 参数指定），导致模型收敛速度不一样，所以我们用两种结构都跑了实验，一般来说在 1x 的 lr schedule 下 Detectron的会高，但 2x 的结果 PyTorch 的结构会比较高。

640?wx_fmt=jpeg

训练速度稍快：Mask R-CNN 差距比较大，其余的很小。采用相同的 setting，Detectron 每个 iteration 需要 0.89s，而 mmdetection 只需要 0.69s。Fast R-CNN 比较例外，比 Detectron 的速度稍慢。另外在我们的服务器上跑 Detectron 会比官方 report 的速度慢 20% 左右，猜测是 FB 的 Big Basin 服务器性能比我们好？

所需显存稍小：显存方面优势比较明显，会小 30% 左右。但这个和框架有关，不完全是 codebase 优化的功劳。一个让我们比较意外的结果是现在的 codebase 版本跑 ResNet-50 的 Mask R-CNN，每张卡（12 G）可以放 4 张图，比我们比赛时候小了不少。

易用性更好：基于 PyTorch 和基于 Caffe2 的 code 相比，易用性是有代差的。成功安装 Detectron 的时间，大概可以装好一打的 mmdetection 吧。

当然，陈恺博士也承认 Detectron 也有一些明显优势，“作为第一个全面的 detection codebase，加上 FAIR 的金字招牌，关注人数和用户很多（虽然吐槽也比较多），release 的模型也比较全面。我们也在努力扩充 model zoo，奈何人力和算力还是有很大差距，所以还需要时间。”

与 mmdetection 一起开源的还有一个基础库——mmcv。据陈恺博士介绍， mmcv 基础库主要分为两个部分：一部分是和 deep learning framework 无关的一些工具函数，比如 IO/Image/Video 相关的一些操作；另一部分是为 PyTorch 写的一套训练工具，可以大大减少用户需要写的代码量，同时让整个流程的定制变得容易。

项目地址：
https://github.com/open-mmlab/mmcv

实际上，mmdetection 和 mmcv 都同属于香港中文大学多媒体实验室的 Open-MMLab 计划。港中大助理教授林达华称，启动 Open-MMLab 计划，是希望在一个统一的代码架构上，逐步开放实验室积累的算法和模型，为计算机视觉的研究社区贡献自己的一分力量。以下是简单说明：

1. 这是一个纯粹的学术开源计划，所开放的都是已经公开发表的算法和模型（包括我们自己和其它研究组提出的代表性算法），不涉及任何的商业化技术。

2. 我们希望这个计划能够降低算法复现的难度，和不必要的重复实验与训练，从而让使用者能够专注于新问题的提出、新思路的探索，而不需要花费主要精力用于已有算法组件的细节调试。

3. 我们在开放的 codebase 中会对现有的主流算法框架进行重构，把它们分解为可复用的组件。希望未来的研究者可以基于这些组件，能迅速建立新的算法框架原型，加快探索的进度。

4. 我们希望 Open-MMLab 成为一个学术交流和探讨的渠道。

参考资料：
https://www.zhihu.com/question/294578141
https://zhuanlan.zhihu.com/p/47011261

【完】

2018 AI开发者大会

只讲技术，拒绝空谈

2018 AI开发者大会是一场由中美人工智能技术高手联袂打造的AI技术与产业的年度盛会！是一场以技术落地为导向的干货会议！大会设置了10场技术专题论坛，力邀15+硅谷实力讲师团和80+AI领军企业技术核心人物，多位一线经验大咖带你将AI从云端落地。

大会日程以及嘉宾议题请查看下方海报（点击查看大图）

https://www.dkcj.cn/info/37283.html

港中大、商汤开源目标检测工具包mmdetection，对比Detectron如何？

相关文章：

关注中国的房地产市场

冠军揭晓！京东Alpha开发者大赛Pick谁上了C位

大白话讲解Promise（二）理解Promise规范

量子计算机人类意识云,意识上传-人类距离永生还要多久(大脑的工作原理很有可能与量子计算机一致)...

Alpha混合

考研计算机专业课怎么复习,2016考研计算机专业课如何复习？

Linux 学习手记(1)：命令行BASH的基本操作

为了智能驾驶，李彦宏要改造城市道路了！

LoadRunner Winsock 10053错误的真正原因

dsp和通用计算机的区别,dsp芯片是什么_dsp芯片和通用微处理器有什么区别

百度大脑发布企业服务解决方案，将 AI 技术落实到细分领域

图片上传(加水印、缩略图、远程保存）的简单例子

wp实例开发精品文章源码推荐

98级计算机应用教材,西安外事学院98级计算机应用9806班毕业二十年校友返校

js：appendChild、insertBefore和insertAfter

来呀！AI喊你斗地主——首个搞定斗地主的深度神经网络

在 Django 的 View 中利用 function decorator 可实现一定程度的代码重用

Linux---管理网络

网络丢包现象分析处理指导书5

信息与计算机科学考研学校排名,计算机科学与技术考研学校排名

800万中文词，腾讯AI Lab开源大规模NLP数据集

屏幕后处理——Bloom

服务端口大全+注解

双绞线是计算机网络的一种通信线路吗,计算机网络环境的信道传输技术分析

为了智能驾驶，李彦宏要改造城市道路

美团大脑：知识图谱的建模方法及其应用 | 公开课笔记

LINQ篇：ASP.NET using LINQ(Part One) Scott大师的产物

大学生如何合理使用计算机,大学生计算机合理使用引导分析

关于机器学习中山峰聚类算法的说明

SQLDMO- （数据备份与恢复篇）