Journal of Computer Applications ›› 2023, Vol. 43 ›› Issue (3): 930-935.DOI: 10.11772/j.issn.1001-9081.2022020168

• Multimedia computing and computer simulation • Previous Articles    

Automatic detection of targets under airport pavement based on channel and spatial attention

Haifeng LI1, Fan ZHANG1, Minnan PIAO1(), Huaichao WANG1, Nansha LI1, Zhongcheng GUI2   

  1. 1.College of Computer Science and Technology,Civil Aviation University of China,Tianjin 300300,China
    2.Chengdu Guimu Robot Company Limited,Chengdu Sichuan 610101,China
  • Received:2022-02-16 Revised:2022-05-12 Accepted:2022-05-13 Online:2022-08-16 Published:2023-03-10
  • Contact: Minnan PIAO
  • About author:LI Haifeng, born in 1984, Ph. D., associate professor. His research interests include robot environmental perception, computer vision.
    ZHANG Fan, born in 1998, M. S. candidate. His research interests include computer vision, image processing.
    WANG Huaichao, born in 1984, Ph. D., lecturer. His research interests include computer vision.
    LI Nansha, born in 1996, Ph. D. candidate. Her research interests include computer vision.
    GUI Zhongcheng, born in 1979, Ph.D., professor of engineering. His research interests include intelligent robot.
  • Supported by:
    National Key Research and Development Program of China(2019YFB1310400);Scientific Research Program of Tianjin Municipal Education Commission(2021KJ036)

基于通道和空间注意力的机场道面地下目标自动检测

李海丰1, 张凡1, 朴敏楠1(), 王怀超1, 李南莎1, 桂仲成2   

  1. 1.中国民航大学 计算机科学与技术学院,天津 300300
    2.成都圭目机器人有限公司,成都 610101
  • 通讯作者: 朴敏楠
  • 作者简介:李海丰(1984—),男,内蒙古通辽人,副教授,博士,CCF会员,主要研究方向:机器人环境感知、计算机视觉
    张凡(1998—),男,四川巴中人,硕士研究生,主要研究方向:计算机视觉、图像处理
    朴敏楠(1993—),女,吉林四平人,讲师,博士,主要研究方向:机器人智能感知与决策
    王怀超(1984—),男,天津人,讲师,博士,主要研究方向:计算机视觉
    李南莎(1996—),女,湖北咸宁人,博士研究生,主要研究方向:计算机视觉
    桂仲成(1979—),男,安徽六安人,教授级高级工程师,博士,主要研究方向:智能机器人。
  • 基金资助:
    国家重点研发计划项目(2019YFB1310400);天津市教委科研计划项目(2021KJ036)

Abstract:

In the task of detecting targets under airport pavement, B-scan maps generated by Ground Penetrating Radar (GPR) have complex backgrounds and lots of noise, especially a single B-scan map cannot reflect the complete information of an underground target. To solve these problems, a Three-Dimensional Channel and Spatial Attention UNet (3D-CSA-UNet) model was established to automatically detect the underground targets. Firstly, a Three-Dimensional Channel and Spatial parallel attention Block (3D-CS-Block) was designed to make the model focus on the underground target information in radar C-scan and suppress the interference of backgrounds and noise. Secondly, in order to enhance the capability of 3D-CS-Block in feature extraction, a multi-scale 3D segmentation model was designed to extract feature maps of different sizes from the radar C-scan. Finally, the cross-entropy loss function was employed to calculate the loss value of feature map under each scale to improve the detection accuracy of the model. On a real dataset of targets under airport pavement, compared with 3D-Fully Convolutional Network (3D-FCN), 3D-UNet and other algorithms, 3D-CSA-UNet has the average F1 score in terms of the pixel level segmentation for void, rebar and parallel rebar targets increased by at last 12.33, 9.05 and 11.05 percentage points. Experimental results show that 3D-CSA-UNet can meet the real engineering requirements well.

Key words: Ground Penetrating Radar (GPR), target detection, Convolutional Neural Network (CNN), channel attention, spatial attention, feature extraction

摘要:

针对机场道面地下目标检测任务中,探地雷达(GPR)生成的B-scan图背景复杂、包含大量噪声,尤其是单个B-scan图不能反映地下目标的完整信息等问题,构建一种三维通道和空间注意力的UNet(3D-CSA-UNet)模型对地下目标进行自动检测。首先,设计三维通道和空间注意力并行模块(3D-CS-Block),使模型重点关注雷达C-scan中的地下目标信息,抑制背景和噪声的干扰;其次,设计多尺度的三维分割模型从雷达C-scan中提取不同大小的特征图,以增强3D-CS-Block提取目标特征的能力;最后,使用交叉熵损失函数计算每个尺度下特征图的损失值,从而提高模型的检测精度。在采集的实际机场道面地下目标数据集上,相较于3D-FCN、3D-UNet等模型,3D-CSA-UNet对于脱空、钢筋和钢筋平行目标预测的平均F1至少提高12.33、9.05、11.05个百分点。实验结果表明,3D-CSA-UNet可以较好地满足工程实际要求。

关键词: 探地雷达, 目标检测, 卷积神经网络, 通道注意力, 空间注意力, 特征提取

CLC Number: