计算机工程与科学2024,Vol.46Issue(1) :91-101.DOI:10.3969/j.issn.1007-130X.2024.01.010

基于改进Deformable DETR的无人机视频流车辆目标检测算法

A vehicle object detection algorithm in UAV video stream based on improved Deformable DETR

江志鹏 王自全 张永生 于英 程彬彬 赵龙海 张梦唯
计算机工程与科学2024,Vol.46Issue(1) :91-101.DOI:10.3969/j.issn.1007-130X.2024.01.010

基于改进Deformable DETR的无人机视频流车辆目标检测算法

A vehicle object detection algorithm in UAV video stream based on improved Deformable DETR

江志鹏 1王自全 1张永生 1于英 1程彬彬 1赵龙海 2张梦唯1
扫码查看

作者信息

  • 1. 战略支援部队信息工程大学地理空间信息学院,河南郑州 450001
  • 2. 32016部队,甘肃兰州 730000
  • 折叠

摘要

针对无人机视频流检测中小目标数量多、因图像传输质量较低而导致的上下文语义信息不充分、传统算法融合特征推理速度慢、数据集类别样本不均衡导致的训练效果差等问题,提出一种基于改进Deformable DETR的无人机视频流车辆目标检测算法.在模型结构方面,该算法设计了跨尺度特征融合模块以增大感受野,提升小目标检测能力,并采用针对object query的挤压-激励模块提升关键目标的响应值,减少重要目标的漏检与错检率;在数据处理方面,使用了在线困难样本挖掘技术,改善数据集中类别样本分布不均的问题.在UAVDT数据集上进行了实验,实验结果表明,改进后的算法相较于基线算法在平均检测精度上提升了 1.5%,在小目标检测精度上提升了 0.8%,并在保持参数量较少增长的情况下,维持了原有的检测速度.

Abstract

Aiming at the problems of a large number of small targets in UAV video stream detection,insufficient contextual semantic information due to low image transmission quality,slow inference speed of traditional algorithm fusion features,and poor training effect caused by unbalanced dataset category samples,this paper proposes a vehicle object detection algorithm based on improved Deformable DETR for UAV video streaming.In terms of model structure,this method designs a cross-scale feature fusion module to increase the receptive field and improve the detection ability of small objects,and adopts the squeeze-excitation module for object_query to improve the response value of key objects and reduce the missed or false detection of important objects.In terms of data processing,online difficult sample min-ing technology is used to improve the problem of uneven distribution of class samples in the data set.The experimental results show that the improved algorithm improves the average detection accuracy by 1.5%and the small target detection accuracy by 1.2%compared with the baseline algorithm without detection speed degradation.

关键词

Deformable/DETR/目标检测/跨尺度特征融合模块/object/query挤压-激励/在线难样本挖掘

Key words

Deformable DETR/object detection/cross-scale feature fusion module/object query squeeze-and-excitation/online hard sample mining

引用本文复制引用

基金项目

国家自然科学基金(42071340)

出版年

2024
计算机工程与科学
国防科学技术大学计算机学院

计算机工程与科学

CSTPCD北大核心
影响因子:0.787
ISSN:1007-130X
参考文献量1
段落导航相关论文