基于迅颈模块与单解耦头的航拍车辆检测方法
Aerial Vehicle Detection Method Based on Fast Neck Module and Single Decoupled Head
高宏伟 1王雨桐1
作者信息
- 1. 沈阳理工大学 自动化与电气工程学院,沈阳 110159
- 折叠
摘要
无人机航拍图像的目标检测需要大量计算资源,导致其在移动端上的应用受到限制,为此提出一种轻量化目标检测网络Faster S.首先,在颈部网络设计了由部分卷积和组卷积组成的轻薄迅颈模块,用以有效聚合不同尺度的信息;其次,在检测头部分设计了一种高效、轻量化的单解耦头输出结构(SDHEAD),用以进一步分离目标检测中分类和定位两个子任务的特征编码.在航拍车辆数据集CARPK上的实验结果表明:与特征金字塔网络(FPN)结构相比,采用迅颈模块提高了模型的检测精度,减少了模型的参数量;相比于其他检测头,SDHEAD具有更强的特征辨别能力;Faster S 模型检测的平均精度(AP0.5)达到了 70.9%,较 Yolo-Fast-estV2 提高了12.7%,较Fastestdet提高了3.5%,推理时间仅为29.56 ms,较Nanodet-m降低了76%,模型参数量仅为0.335 M.
Abstract
Object detection in drone aerial images require intensive computational resources,which limits the application on mobile terminals.Therefore,a lightweight target detection network Faster S is proposed.Firstly,a light and thin neck module composed of partial convolution and group convo-lution is designed in the neck network to effectively aggregate information of different scales.Sec-ondly,an efficient and lightweight single decoupled head output structure(SDHEAD)is designed in the detection head part,which can further separate the feature encoding of the two subtasks of clas-sification and localization in object detection.The results of experimental on the aerial vehicle data-set CARPK show that compared with the feature pyramid network(FPN)structure,the fast neck module improves the detection accuracy of the model and reduces the parameters of the model.Compared with other detection heads,SDHEAD has stronger feature discrimination ability.The av-erage accuracy(AP0.5)of Faster S model detection reaches 70.9%,which is 12.7%higher than Yolo-FastestV2,3.5%higher than Fastestdet,and the inference time is only 29.56 ms,which is 76%lower than Nanodet-m.The parameter size is only 0.335 M.
关键词
目标检测/航拍图像/轻量化/单解耦头/迅颈模块Key words
object detection/aerial image/lightweight/single decoupled head/fast neck module引用本文复制引用
基金项目
辽宁省重点科技创新基地联合开放基金(2021-KF-12-05)
出版年
2024
NETL
NSTL
万方数据