HyperAIHyperAI

Command Palette

Search for a command to run...

RTMDet:实时目标检测器设计的实证研究

Chengqi Lyu Wenwei Zhang Haian Huang Yue Zhou Yudong Wang Yanyi Liu Shilong Zhang Kai Chen

摘要

在本文中,我们旨在设计一种高效的实时目标检测器,其性能超越YOLO系列,并且易于扩展到多种目标识别任务,如实例分割和旋转目标检测。为了获得更高效模型架构,我们探索了一种在主干网络和颈部具有兼容容量的架构,该架构由包含大核深度卷积的基本构建块组成。此外,我们在动态标签分配计算匹配成本时引入了软标签以提高精度。结合更好的训练技术,所得到的目标检测器命名为RTMDet,在NVIDIA 3090 GPU上实现了超过300帧每秒(FPS)的52.8% COCO平均精度(AP),优于当前主流的工业级检测器。RTMDet在不同应用场景下通过微小/小型/中型/大型/超大型模型尺寸实现了最佳的参数-精度权衡,并在实时实例分割和旋转目标检测方面取得了新的最先进性能。我们希望实验结果能为设计适用于多种目标识别任务的多功能实时目标检测器提供新的见解。代码和模型已发布在https://github.com/open-mmlab/mmdetection/tree/3.x/configs/rtmdet


用 AI 构建 AI

从创意到上线——通过免费 AI 协同编码、开箱即用的环境和最优惠的 GPU 价格,加速您的 AI 开发。

AI 协同编码
开箱即用的 GPU
最优定价

HyperAI Newsletters

订阅我们的最新资讯
我们会在北京时间 每周一的上午九点 向您的邮箱投递本周内的最新更新
邮件发送服务由 MailChimp 提供