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

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

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

主要特性:

  • 模块化设计

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

  • 支持多个框架开箱即用

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

  • 高效

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

  • 最先进的

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

浏览 5
点赞
评论
收藏
分享

手机扫一扫分享

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

手机扫一扫分享

编辑
举报