Detection method of remote sensing image ship based on YOLOv5
DOI:
CSTR:
Author:
Affiliation:

1.Nanjing University Of Information Science & Technology, Nanjing 210044, ,China;2.Nanjing University Of Information Science & Technology, Jiangsu Collaborative Innovation Center of Atmospheric Environment and Equipment Technology,Nanjing210044, ,China; 3. Binjiang College of,Nanjing University of Information Science &TechnologyWuxi 214105,China;4. Shanghai Satellite Engineering Research Institute, Shanghai 201100,China

Clc Number:

TP391

Fund Project:

  • Article
  • |
  • Figures
  • |
  • Metrics
  • |
  • Reference
  • |
  • Related
  • |
  • Cited by
  • |
  • Materials
  • |
  • Comments
    Abstract:

    The use of remote sensing images to monitor ships on the sea has become a hot spot in current research. In order to solve the defects of traditional ship detection that requires manual feature selection, time-consuming and labor-consuming, and the original YOLO algorithm has low detection accuracy for densely distributed small targets. This paper proposes a remote sensing image ship detection method based on YOLOv5, using the remote sensing data set provided by the Kaggle platform, training on the Pytorch framework, the loss function is designed as CIOU_LOSS, and the selection of the target frame uses the DIOU_NMS algorithm to make the occluded and overlapped The target detection effect is enhanced. After experimental comparison, the detection accuracy of this target detection model for occluded and densely arranged ships is better than other models, and its average detection accuracy is increased from the original 88.75% to 91.27%.

    Reference
    Related
    Cited by
Get Citation
Share
Article Metrics
  • Abstract:
  • PDF:
  • HTML:
  • Cited by:
History
  • Received:
  • Revised:
  • Adopted:
  • Online: October 11,2024
  • Published:
Article QR Code