摘要
本发明涉及一种基于双预测头高精度轻量化的无人机图像目标检测算法,结合无人机图像目标检测的任务特点进行优化和改进,提出一种双目标检测头模块,该模块可以在提高小目标检测精度同时还能降低模型的参数量。此外,本发明进一步设计一个小路径特征融合模块(Tiny Path Fusion,TP‑Fusion),该模块能融合更丰富的细节信息,并使检测器能更准确地匹配不同大小和形状的目标。最后,本发明进一步采用坐标注意力模块(Coordinate Attention Block,CA)和改进的瓶颈模块,降低复杂的背景对检测前景目标信息干扰,增强了无人机图像小目标检测。
技术关键词
无人机航拍图像
算法
注意力
卷积模块
路径特征
全局平均池化
网络
基线
批量
分辨率
输出特征
坐标
定义
瓶颈
检测头
通道
检测器