lyuwenyu

RT-DETR

lyuwenyu

[CVPR 2024] Official RT-DETR (RTDETR paddle pytorch), Real-Time DEtection TRansformer, DETRs Beat YOLOs on Real-time Object Detection. 🔥 🔥 🔥

AI 简介

RT-DETR 是一个面向实时目标检测的端到端 Transformer 模型开源实现,支持 PyTorch 和 PaddlePaddle 双框架。其核心特点包括:基于 DETR 架构的低延迟设计、无需 NMS 后处理、支持多尺度特征融合与动态查询机制,并提供 RT-DETRv2 等持续演进版本,兼顾高精度与推理速度(如 RT-DETRv2-S 达 48.1 mAP)。项目内置训练/推理/评估全流程工具,支持 sliced inference(小目标检测)、模型分析(参数/FLOPs 统计)及 Torch Hub 集成。适用于对延迟敏感的工业质检、智能安防、边缘部署等实时视觉感知场景。

Python
Apache License 2.0
5.3k
Stars
636
Forks
36
Watchers
412
Issues

Star 增长

今日0
近 7 天0
近 30 天+21
综合评分66.51
默认分支main