发明公开
- 专利标题: 一种基于改进DAB-DETR的目标检测方法及系统
-
申请号: CN202410049017.X申请日: 2024-01-12
-
公开(公告)号: CN117876775A公开(公告)日: 2024-04-12
- 发明人: 张创 , 桂彦
- 申请人: 长沙理工大学
- 申请人地址: 湖南省长沙市天心区万家丽南路二段960号
- 专利权人: 长沙理工大学
- 当前专利权人: 长沙理工大学
- 当前专利权人地址: 湖南省长沙市天心区万家丽南路二段960号
- 主分类号: G06V10/764
- IPC分类号: G06V10/764 ; G06V10/25 ; G06V10/82 ; G06N3/0455 ; G06N3/08
摘要:
本发明公开了一种基于改进DAB‑DETR的目标检测方法及系统,基于DAB‑DETR结构改进而来,在不改变原有结构的基础上,开发了两种训练设计,有效解决了查询之间的冗余预测以及分类置信度和定位精度之间的不对齐问题。具体而言,我们提出了一种查询感知的区域约束,通过限制每个查询的责任区域,避免了查询之间不必要的内部冲突。然后,我们引入了一种位置调制的分类损失,通过加强分类置信度与定位精度之间的相关性,鼓励模型产生分类置信度对齐定位精度的预测。提出的两种设计极大提高了DAB‑DETR的收敛速度和检测精度,并且没有增加模型的参数量和计算成本,可以很容易集成到现有基于DETR的模型,带来进一步的提升。