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