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 条
  • [21] Remote Distance Binocular Vision Ranging Method Based on Improved YOLOv5
    Wei, Biaobiao
    Liu, Jun
    Li, Ao
    Cao, Huiliang
    Wang, Chenguang
    Shen, Chong
    Tang, Jun
    IEEE SENSORS JOURNAL, 2024, 24 (07) : 11328 - 11341
  • [22] Improved YOLOv5 Object Detection Algorithm for Remote Sensing Images
    Yang, Chen
    She, Lu
    Yang, Lu
    Feng, Zixian
    Computer Engineering and Applications, 2023, 59 (15) : 76 - 86
  • [23] Ship target detection algorithm of optical remote sensing image based on YOLOv5
    Cheng Q.
    Li J.
    Du J.
    Xi Tong Gong Cheng Yu Dian Zi Ji Shu/Systems Engineering and Electronics, 2023, 45 (05): : 1270 - 1276
  • [24] Improved YOLOv5 in Remote Sensing Slender and Rotating Target Detection
    Zhong Bo
    Yang Luyuan
    2022 IEEE INTERNATIONAL CONFERENCE ON ELECTRICAL ENGINEERING, BIG DATA AND ALGORITHMS (EEBDA), 2022, : 918 - 923
  • [25] Strawberry Maturity Recognition Based on Improved YOLOv5
    Tao, Zhiqing
    Li, Ke
    Rao, Yuan
    Li, Wei
    Zhu, Jun
    AGRONOMY-BASEL, 2024, 14 (03):
  • [26] Surgical Instrument Recognition Based on Improved YOLOv5
    Jiang, Kaile
    Pan, Shuwan
    Yang, Luxuan
    Yu, Jie
    Lin, Yuanda
    Wang, Huaiqian
    APPLIED SCIENCES-BASEL, 2023, 13 (21):
  • [27] A Vehicle Recognition Model Based on Improved YOLOv5
    Shao, Lei
    Wu, Han
    Li, Chao
    Li, Ji
    ELECTRONICS, 2023, 12 (06)
  • [28] Improved Fisheye Image Target Detection Algorithm Based on YOLOv5 Network
    Lyu, Xiaoling
    Yang, Shengyue
    Zhang, Minglu
    Liang, Ming
    Wang, Junchao
    Computer Engineering and Applications, 2023, 59 (06): : 241 - 250
  • [29] Remote Sensing Infrared Weak and Small Target Detection Method Based on Improved YOLOv5 and Data Augmentation
    Zhang, Meixin
    Liu, Zhonghua
    Zhang, Peng
    Yu, Qian
    Li, Zhiyuan
    Li, Yi
    INTELLIGENT ROBOTICS AND APPLICATIONS, ICIRA 2024, PT IX, 2025, 15209 : 312 - 324
  • [30] Apple rapid recognition and processing method based on an improved version of YOLOv5
    Wang, Jiuxin
    Su, Yaoheng
    Yao, Jiahui
    Liu, Man
    Du, Yurong
    Wu, Xin
    Huang, Lei
    Zhao, Minghu
    ECOLOGICAL INFORMATICS, 2023, 77