A Vehicle Detection Method Based on an Improved U-YOLO Network for High-Resolution Remote-Sensing Images

被引:8
作者
Guo, Dudu [1 ,2 ]
Wang, Yang [2 ]
Zhu, Shunying [1 ]
Li, Xin [2 ]
机构
[1] Wuhan Univ Technol, Sch Transportat & Logist Engn, Wuhan 430070, Peoples R China
[2] Xinjiang Univ, Coll Transportat Engn, Urumqi 830046, Peoples R China
关键词
U-YOLO; cross-scale channel attention; remote-sensing images; vehicle inspection;
D O I
10.3390/su151310397
中图分类号
X [环境科学、安全科学];
学科分类号
08 ; 0830 ;
摘要
The lack of vehicle feature information and the limited number of pixels in high-definition remote-sensing images causes difficulties in vehicle detection. This paper proposes U-YOLO, a vehicle detection method that integrates multi-scale features, attention mechanisms, and sub-pixel convolution. The adaptive fusion module (AF) is added to the backbone of the YOLO detection model to increase the underlying structural information of the feature map. Cross-scale channel attention (CSCA) is introduced to the feature fusion part to obtain the vehicle's explicit semantic information and further refine the feature map. The sub-pixel convolution module (SC) is used to replace the linear interpolation up-sampling of the original model, and the vehicle target feature map is enlarged to further improve the vehicle detection accuracy. The detection accuracies on the open-source datasets NWPU VHR-10 and DOTA were 91.35% and 71.38%. Compared with the original network model, the detection accuracy on these two datasets was increased by 6.89% and 4.94%, respectively. Compared with the classic target detection networks commonly used in RFBnet, M2det, and SSD300, the average accuracy rate values increased by 6.84%, 6.38%, and 12.41%, respectively. The proposed method effectively solves the problem of low vehicle detection accuracy. It provides an effective basis for promoting the application of high-definition remote-sensing images in traffic target detection and traffic flow parameter detection.
引用
收藏
页数:15
相关论文
共 24 条
  • [1] Improved YOLOv3 Based on Attention Mechanism for Fast and Accurate Ship Detection in Optical Remote Sensing Images
    Chen, Liqiong
    Shi, Wenxuan
    Deng, Dexiang
    [J]. REMOTE SENSING, 2021, 13 (04) : 1 - 18
  • [2] Small object detection in remote sensing images based on super-resolution
    Fang Xiaolin
    Hu Fan
    Yang Ming
    Zhu Tongxin
    Bi Ran
    Zhang Zenghui
    Gao Zhiyuan
    [J]. PATTERN RECOGNITION LETTERS, 2022, 153 : 107 - 112
  • [3] A Slimmer Network with Polymorphic and Group Attention Modules for More Efficient Object Detection in Aerial Images
    Guo, Wei
    Li, Weihong
    Li, Zhenghao
    Gong, Weiguo
    Cui, Jinkai
    Wang, Xinran
    [J]. REMOTE SENSING, 2020, 12 (22) : 1 - 30
  • [4] Deep Residual Learning for Image Recognition
    He, Kaiming
    Zhang, Xiangyu
    Ren, Shaoqing
    Sun, Jian
    [J]. 2016 IEEE CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR), 2016, : 770 - 778
  • [5] A Light-Weight Neural Network Using Multiscale Hybrid Attention for Building Change Detection
    Hua, Zhihua
    Yu, Haiyang
    Jing, Peng
    Song, Caoyuan
    Xie, Saifei
    [J]. SUSTAINABILITY, 2023, 15 (04)
  • [6] An Efficient Multilevel Probabilistic Model for Abnormal Traffic Detection in Wireless Sensor Networks
    Khan, Muhammad Altaf
    Nasralla, Moustafa M.
    Umar, Muhammad Muneer
    Ghani-Ur-Rehman
    Khan, Shafiullah
    Choudhury, Nikumani
    [J]. SENSORS, 2022, 22 (02)
  • [7] Li X., 2021, P INT C COMPUTER VIS, VVolume 12155, P141
  • [8] Improved YOLOv3 model for vehicle detection in high-resolution remote sensing images
    Li, Yuntao
    Wu, Zhihuan
    Li, Lei
    Yang, Daoning
    Pang, Hongfeng
    [J]. JOURNAL OF APPLIED REMOTE SENSING, 2021, 15 (02)
  • [9] Remote Sensing Image Vehicle Detection Based on Pre-Training and Random-Initialized Fusion Network
    Liu, Hongkun
    Ding, Qichen
    Hu, Zican
    Chen, Xueyun
    [J]. IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2022, 19
  • [10] Receptive Field Block Net for Accurate and Fast Object Detection
    Liu, Songtao
    Huang, Di
    Wang, Yunhong
    [J]. COMPUTER VISION - ECCV 2018, PT XI, 2018, 11215 : 404 - 419