mmdetection 介绍
mmdetection 是一个基于 PyTorch
的开源对象检测工具箱,它提供了已公开发表的多种视觉检测核心模块,通过这些模块的组合,可以迅速搭建出各种著名的检测框架。
主要特性:
- 模块化设计
通过组合不同的组件,可以容易地构建定制的对象检测框架。
- 支持多个框架开箱即用
该工具箱直接支持流行的检测框架,如 Faster RCNN、Mask RCNN、RetinaNet 等。
- 高效
所有基本的 bBox 和掩码操作现在都在 GPU 上运行。不同模型上,训练速度比 Detectron 快约 5%~20%。
- 最先进的
重构自 MMDet 团队的代码库,而 MMDet 赢得了 COCO Detection 2018 挑战赛。