mmdetection - 基于 PyTorch 的对象检测工具箱


GPL-3.0
跨平台
Python

软件简介

mmdetection 是一个基于 PyTorch
的开源对象检测工具箱,它提供了已公开发表的多种视觉检测核心模块,通过这些模块的组合,可以迅速搭建出各种著名的检测框架。

主要特性:

  • 模块化设计

通过组合不同的组件,可以容易地构建定制的对象检测框架。

  • 支持多个框架开箱即用

该工具箱直接支持流行的检测框架,如 Faster RCNN、Mask RCNN、RetinaNet 等。

  • 高效

所有基本的 bbox 和掩码操作现在都在 GPU 上运行。不同模型上,训练速度比 Detectron 快约 5%~20%。

  • 最先进的

重构自 MMDet 团队的代码库,而 MMDet 赢得了 COCO Detection 2018 挑战赛。