Improved YOLOv5 network method for remote sensing image-based ground objects recognition

被引:30
|
作者
Xue, Jie [1 ]
Zheng, Yongguo [1 ]
Dong-Ye, Changlei [1 ]
Wang, Ping [2 ]
Yasir, Muhammad [3 ]
机构
[1] Shandong Univ Sci & Technol, Coll Comp Sci & Engn, Qingdao 266590, Shandong, Peoples R China
[2] Shandong Univ Sci & Technol, Coll Geodesy & Geomat, Qingdao 266590, Shandong, Peoples R China
[3] China Univ Petr East China, Coll Oceanog & Space Informat, Qingdao 266580, Shandong, Peoples R China
关键词
High-resolution remote sensing image; Object recognition; YOLOv5; Attention mechanism; CLASSIFICATION;
D O I
10.1007/s00500-022-07106-8
中图分类号
TP18 [人工智能理论];
学科分类号
081104 ; 0812 ; 0835 ; 1405 ;
摘要
High-resolution remote sensing images have the characteristics of complex background environment, clustering of objects, etc., the complex background makes the remote sensing image contain a large number of irrelevant ground objects with a high similarity or overlap, which makes the edge and texture of the objects not clear enough, and this leads to low recognition accuracy of ground objects such as airports, dams, and golf field, although the size of this object is large. Based on this problem, this paper proposes a remote sensing image object detection method based on the YOLOv5 network. By improving the backbone extraction network, the network structure can be deepened to get more information about large objects, and the detection effect can be improved by adding an attention mechanism and adding an output layer to enhance feature extraction and feature fusion. The pre-training weight is obtained by transfer learning and used as the training weight of the improved YOLOv5 to speed up the network convergence. The experiment is carried out on the DIOR dataset, the results show that the improved YOLOv5 network can significantly improve the accuracy of large object recognition compared with the YOLO series network and the EfficientDet model on DIOR dataset, and the mAP of the improved YOLOv5 network is 80.5%, which is 2% higher than the original YOLOv5 network.
引用
收藏
页码:10879 / 10889
页数:11
相关论文
共 50 条
  • [1] Improved YOLOv5 network method for remote sensing image-based ground objects recognition
    Jie Xue
    Yongguo Zheng
    Changlei Dong-Ye
    Ping Wang
    Muhammad Yasir
    Soft Computing, 2022, 26 : 10879 - 10889
  • [2] Improved YOLOv5 for Remote Sensing Image Detection
    Liu, Tao
    Ding, Xueyan
    Zhang, Bingbing
    Zhang, Jianxin
    Computer Engineering and Applications, 2023, 59 (10): : 253 - 261
  • [3] Remote Sensing Image Target Detection and Recognition Based on YOLOv5
    Liu, Xiaodong
    Gong, Wenyin
    Shang, Lianlian
    Li, Xiang
    Gong, Zixiang
    REMOTE SENSING, 2023, 15 (18)
  • [4] Target Detection of Remote Sensing Image Based on an Improved YOLOv5
    Han, Hao
    Zhu, Fuzhen
    Zhu, Bing
    Wu, Hong
    IEEE GEOSCIENCE AND REMOTE SENSING LETTERS, 2023, 20
  • [5] Optical Remote Sensing Ship Recognition and Classification Based on Improved YOLOv5
    Jian, Jun
    Liu, Long
    Zhang, Yingxiang
    Xu, Ke
    Yang, Jiaxuan
    REMOTE SENSING, 2023, 15 (17)
  • [6] Target Detection Algorithm of Remote Sensing Image Based on Improved YOLOv5
    Li, Kunya
    Ou, Ou
    Liu, Guangbin
    Yu, Zefeng
    Li, Lin
    Computer Engineering and Applications, 2023, 59 (09) : 207 - 214
  • [7] Improved YOLOv4 Remote Sensing Image Detection Method of Ground Objects Along Railway
    Wang Yang-ping
    Han Shu-mei
    Yang Jing-yu
    Dang Jian-wu
    Zhang Zhan-ping
    SPECTROSCOPY AND SPECTRAL ANALYSIS, 2022, 42 (10) : 3275 - 3282
  • [8] An Improved Lightweight YOLOv5 for Remote Sensing Images
    Hou, Shihao
    Fan, Linwei
    Zhang, Fan
    Liu, Bingchen
    ARTIFICIAL NEURAL NETWORKS AND MACHINE LEARNING, ICANN 2023, PT II, 2023, 14255 : 77 - 89
  • [9] A rotatable battery recognition method based on improved YOLOv5
    Chen, Wenming
    Liang, Dongtai
    Ding, Wenhui
    Wang, Meng
    Chen, Zizhen
    INTERNATIONAL JOURNAL OF SENSOR NETWORKS, 2024, 45 (02) : 101 - 114
  • [10] Fish sonar image recognition algorithm based on improved YOLOv5
    Xing, Bowen
    Sun, Min
    Ding, Minyang
    Han, Chuang
    MATHEMATICAL BIOSCIENCES AND ENGINEERING, 2024, 21 (01) : 1321 - 1341