商汤的mmdetection3D检测框架开源啦
文档代码地址
Github: https://github.com/open-mmlab/mmdetection3d
文档: https://mmdetection3d.readthedocs.io/
框架特点
- 支持多模态/单模态检测器
它直接支持多模态/单模态检测器,包括MVXNet,VoteNet,PointPillars等。- 支持开箱即用的室内/室外3D检测
它直接支持流行的室内和室外3D检测数据集,包括ScanNet,SUNRGB-D,nuScenes,Lyft和KITTI。- 与2D检测自然融合
MMDetection支持的所有大约40多种方法,300多种模型和模块都可以在此代码库中进行训练或使用。- 高效率
它比其他代码库的训练速度更快。主要结果如下。详细信息可以在Benchmark.md中找到。我们比较每秒训练的样本数量(越高,越好)。其他代码库不支持的模型用标记×。
像MMDetection和MMCV一样,MMDetection3D也可以用作支持其上不同项目的库。
Benchmark and model zoo
下表显示了受支持的方法和主干。结果和模型可在model zoo获得。
注意:MMDetection支持的所有大约300种模型, 2D检测中40多种论文的方法都可以在此代码库中进行训练或使用。
最后
MMDetection3D是一个开放源代码项目,由来自各个学院和公司的研究人员和工程师提供。我们感谢所有贡献者以及提供宝贵反馈的用户。我们希望该工具箱和基准可以通过提供灵活的工具箱来重新实现现有方法并开发自己的新3D检测器,从而为不断发展的研究社区服务。