Target detection tasks on water have been affected by water surface fluctuations, light changes, occlusions, overlaps and targets that are too small, resulting in poor recognition accuracy and detection speed. To solve this problem, this paper proposes an improved water target detection algorithm based on YOLOv5. Firstly, recursive gated convolution module were used instead of ordinary convolution to achieve high-order spatial interactions and improve model performance and confidence. Secondly, in order to solve the problems of difficulty in detecting small targets, the input size of the image is increased and P6 detection heads are introduced to improve the detection accuracy of small targets. Finally, to alleviate the situation of poor overlapping target detection, coordinate attention(CA) mechanism is introduced and incorporated into the backbone to extract large range features, and reduce the number of parameters and computational overhead. Compared with the original model of YOLOv5s, the improved algorithm model improves the recognition accuracy by 4.6% and the recall by 3.4%, reaching 69.2% and 67.6% respectively. The running speed reaches 208FPS, meeting the requirements of real-time detection.
Access to the requested content is limited to institutions that have purchased or subscribe to SPIE eBooks.
You are receiving this notice because your organization may not have SPIE eBooks access.*
*Shibboleth/Open Athens users─please
sign in
to access your institution's subscriptions.
To obtain this item, you may purchase the complete book in print or electronic format on
SPIE.org.
INSTITUTIONAL Select your institution to access the SPIE Digital Library.
PERSONAL Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.