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

联合创作 · 2023-09-26 04:47

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

主要特性:

  • 模块化设计

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

  • 支持多个框架开箱即用

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

  • 高效

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

  • 最先进的

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

浏览 17
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报
评论
图片
表情
推荐
点赞
评论
收藏
分享

手机扫一扫分享

编辑 分享
举报