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 条
  • [31] A coral reef benthic recognition method based on the improved YOLOv5 algorithm
    Wu, Rui
    Bi, Xiaojun
    Harbin Gongcheng Daxue Xuebao/Journal of Harbin Engineering University, 2022, 43 (04): : 580 - 586
  • [32] Research on broiler health status recognition method based on improved YOLOv5
    Cui, Yue
    Kong, Xiangchao
    Chen, Changxi
    Li, Yiwei
    SMART AGRICULTURAL TECHNOLOGY, 2023, 6
  • [33] Research on Fine-Grained Image Recognition of Birds Based on Improved YOLOv5
    Yi, Xiaomei
    Qian, Cheng
    Wu, Peng
    Maponde, Brian Tapiwanashe
    Jiang, Tengteng
    Ge, Wenying
    SENSORS, 2023, 23 (19)
  • [34] Improved YOLOv5 Gesture Recognition Method in Complex Environments
    Yan, Haoyue
    Wang, Wei
    Tian, Ze
    Computer Engineering and Applications, 2024, 59 (04) : 224 - 234
  • [35] Multi-branch stacking remote sensing image target detection based on YOLOv5
    Bian, Luxuan
    Li, Bo
    Wang, Jue
    Gao, Zijun
    EGYPTIAN JOURNAL OF REMOTE SENSING AND SPACE SCIENCES, 2023, 26 (04): : 999 - 1008
  • [36] UAV Recognition and Tracking Method Based on YOLOv5
    Bie, Tong
    Fan, Kuangang
    Tang, Yaofeng
    2022 IEEE 17TH CONFERENCE ON INDUSTRIAL ELECTRONICS AND APPLICATIONS (ICIEA), 2022, : 1234 - 1239
  • [37] Lymphocyte Detection Method Based on Improved YOLOv5
    Jiang, Peihe
    Li, Yi
    Liu, Ying
    Lu, Ning
    IEEE ACCESS, 2024, 12 : 772 - 781
  • [38] Lane Line Type Recognition Based on Improved YOLOv5
    Liu, Boyu
    Wang, Hao
    Wang, Yongqiang
    Zhou, Congling
    Cai, Lei
    APPLIED SCIENCES-BASEL, 2023, 13 (18):
  • [39] Lightweight Sea Cucumber Recognition Network Using Improved YOLOv5
    Xiao, Qian
    Li, Qian
    Zhao, Lide
    IEEE ACCESS, 2023, 11 : 44787 - 44797
  • [40] Fish detection method based on improved YOLOv5
    Li, Lei
    Shi, Guosheng
    Jiang, Tao
    AQUACULTURE INTERNATIONAL, 2023, 31 (05) : 2513 - 2530