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