基于改进Transformer的小目标车辆精确检测算法

摘要：

智能交通系统的建立离不开车辆检测技术。目前的主流方案是使用卷积神经网络（CNN）架构进行车辆检测，然而在复杂交通场景中，远距离小目标像素点少，CNN的下采样机制导致提取的特征缺乏充足的上下文信息，因而小目标检测面临极大挑战。针对这个问题，提出了一种基于视觉Transformer的小目标车辆检测算法。所提算法通过改进Transformer的线性嵌入模块，补充小目标的线性嵌入信息；对图像进行层级构建，每层仅对局部进行关系建模，同时扩大感受野，代替CNN提取出更强有力的小目标车辆特征，实现端到端的精确检测。在UA-DETRAC车辆数据集上进行验证，实验结果表明，改进后的车辆检测算法提高了对远距离及严重遮挡情况下小目标的检测性能，检测精度达到99.0%。

关键词：机器视觉、车辆检测、小目标、图像增强、视觉Transformer

所属期刊栏目：59

分类号：TP391.41;U495;TN911.73

在线出版日期：2022-10-17（万方平台首次上网日期，不代表论文的发表时间）

页数：共1页

页码：1815016