1) RT-DETR neck代替YOLOv5 neck部分; 2)引入RTDETRDecoder 1.RT-DETR介绍 论文:https://arxiv.org/pdf/2304.08069.pdf RT-DETR(Real-TimeDEtectionTRansformer) ,一种基于 DETR 架构的实时端到端检测器,其在速度和精度上取得了 SOTA 性能 为什么会出现: YOLO 检测器有个较大的待改进点是需要 NMS ...
RT-DETR号称是打败YOLO的检测模型,其作为一种基于Transformer的检测方法,相较于传统的基于卷积的检测方法,提供了更为全面和深入的特征理解,将RT-DETR中的一些先进模块融入到YOLOv5往往能够达到一些特殊的效果(我个人猜测,所以我进行了一些实验来验证这一点),我将RT-DETR的AIFI模块和Conv模块融合在一起添加到了YOLOv5...