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 条
  • [41] Recognition new energy vehicles based on improved YOLOv5
    Hu, Yannan
    Kong, Mingming
    Zhou, Mingsheng
    Sun, Zhanbo
    FRONTIERS IN NEUROROBOTICS, 2023, 17
  • [42] Research on Gesture Recognition Based on Improved YOLOv5 and Mediapipe
    Ni, Guangxing
    Xu, Hua
    Wang, Chao
    Computer Engineering and Applications, 60 (07): : 108 - 118
  • [43] Grazing Sheep Behaviour Recognition Based on Improved YOLOV5
    Hu, Tianci
    Yan, Ruirui
    Jiang, Chengxiang
    Chand, Nividita Varun
    Bai, Tao
    Guo, Leifeng
    Qi, Jingwei
    SENSORS, 2023, 23 (10)
  • [44] Fish detection method based on improved YOLOv5
    Lei Li
    Guosheng Shi
    Tao Jiang
    Aquaculture International, 2023, 31 : 2513 - 2530
  • [45] Helmet detection method based on improved YOLOv5
    Hou G.
    Chen Q.
    Yang Z.
    Zhang Y.
    Zhang D.
    Li H.
    Gongcheng Kexue Xuebao/Chinese Journal of Engineering, 2024, 46 (02): : 329 - 342
  • [46] Faster and Lightweight: An Improved YOLOv5 Object Detector for Remote Sensing Images
    Zhang, Jiarui
    Chen, Zhihua
    Yan, Guoxu
    Wang, Yi
    Hu, Bo
    REMOTE SENSING, 2023, 15 (20)
  • [47] An Improved UAV Detection Method Based on YOLOv5
    Liu, Xinfeng
    Chen, Mengya
    Li, Chenglong
    Tian, Jie
    Zhou, Hao
    Ullah, Inam
    ADVANCED INTELLIGENT COMPUTING TECHNOLOGY AND APPLICATIONS, ICIC 2023, PT I, 2023, 14086 : 739 - 750
  • [48] Underwater Waste Recognition and Localization Based on Improved YOLOv5
    Niu J.
    Gu S.
    Du J.
    Hao Y.
    Computers, Materials and Continua, 2023, 76 (02): : 2015 - 2031
  • [49] Tomato recognition and location algorithm based on improved YOLOv5
    Li, Tianhua
    Sun, Meng
    He, Qinghai
    Zhang, Guanshan
    Shi, Guoying
    Ding, Xiaoming
    Lin, Sen
    COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2023, 208
  • [50] Underwater Waste Recognition and Localization Based on Improved YOLOv5
    Niu, Jinxing
    Gu, Shaokui
    Du, Junmin
    Hao, Yongxing
    CMC-COMPUTERS MATERIALS & CONTINUA, 2023, 76 (02): : 2015 - 2031