一种基于改进DAB-DETR的目标检测方法及系统
摘要:
本发明公开了一种基于改进DAB‑DETR的目标检测方法及系统,基于DAB‑DETR结构改进而来,在不改变原有结构的基础上,开发了两种训练设计,有效解决了查询之间的冗余预测以及分类置信度和定位精度之间的不对齐问题。具体而言,我们提出了一种查询感知的区域约束,通过限制每个查询的责任区域,避免了查询之间不必要的内部冲突。然后,我们引入了一种位置调制的分类损失,通过加强分类置信度与定位精度之间的相关性,鼓励模型产生分类置信度对齐定位精度的预测。提出的两种设计极大提高了DAB‑DETR的收敛速度和检测精度,并且没有增加模型的参数量和计算成本,可以很容易集成到现有基于DETR的模型,带来进一步的提升。
0/0