HyperAIHyperAI

Command Palette

Search for a command to run...

Console
6 个月前

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

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

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

摘要

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

代码仓库

yxb-nku/strip-r-cnn
pytorch
GitHub 中提及
cszzshi/SimD
pytorch
GitHub 中提及
HVision-NKU/Strip-R-CNN
pytorch
GitHub 中提及
open-mmlab/mmyolo
pytorch
GitHub 中提及
open-mmlab/mmrotate
pytorch
GitHub 中提及
yuyi1005/point2rbox-mmrotate
pytorch
GitHub 中提及
fiveai/MoCaE
pytorch
GitHub 中提及
RangiLyu/mmdetection_test
pytorch
GitHub 中提及
V3Det/mmdetection-V3Det
pytorch
GitHub 中提及
open-edge-platform/geti
pytorch
GitHub 中提及

用 AI 构建 AI

从想法到上线——通过免费 AI 协同编程、开箱即用的环境和市场最优价格的 GPU 加速您的 AI 开发

AI 协同编程
即用型 GPU
最优价格
立即开始

Hyper Newsletters

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