特征信息增强的无人机车辆实时检测算法

Real-Time UAV Vehicle Detection Based on Enhanced Feature Information

  • 摘要: 针对无人机视角下车辆由于尺度小分辨率低等问题而难以精确分类定位,本文设计了一个轻量级特征提取网络用于提供车辆的多尺度中低层信息,并分别将其融入到主干神经网络中,实现中低层特征信息的传递;同时利用主干网络提取有利于车辆与背景或其他类别分类的高级语义信息,然后将深层高级语义特征与浅层特征进行融合实现高级语义信息的传递,因此类似引入双向网络能够有效地传递不同层次的信息,增强车辆的特征信息表示。此外,采用多路空洞卷积进行特征提取,使得中低层信息更加丰富多样性;并设计了一种灵活有效的融合模块,能够将中低层信息较好地融入到主干网络中增强目标车辆的判别性特征。实验结果表明,该算法能够在无人机数据集上取得很好的检测效果,同样满足实时的应用需求。

     

    Abstract: ‍ ‍Vehicles from unmanned aerial vehicle (UAV) images were difficult to achieve accurate classification and location, due to the objects were small size and low-resolution. A light-weight feature extraction network was designed in this paper to provide multi-scale mid-/low-level feature that was integrated into the backbone network, which realized the transmission of mid-/low-level information. At the same time, the high-level semantic information was extracted from the backbone, which was beneficial to differentiate the target vehicle from background or other vehicle categories, then deep high-level semantic features and shallow features were fused to realize the transmission of high-level information. Thus, a similar bi-directional network was introduced that could effectively transfer information from different levels and enhance the feature representation for vehicles. Furthermore, multi-rate dilated convolution was proposed to obtain richer mid-/low-level information, and an effective feature fusion module was presented to integrate the mid-/low-level information into the backbone. The experimental results showed that the proposed algorithm could achieve accurate classification and location for UAV vehicles and realize the real-time application requirements.

     

/

返回文章
返回