Vessel classification algorithm based on the convolutional attention module for natural maritime images

Shi-Jia Zhao; Jia-Chun Zheng; Lei Zhang; Yan-duo Hu

doi:10.1117/12.2680442

27 June 2023 Vessel classification algorithm based on the convolutional attention module for natural maritime images

Shi-Jia Zhao, Jia-Chun Zheng, Lei Zhang, Yan-duo Hu

Proceedings Volume 12705, Fourteenth International Conference on Graphics and Image Processing (ICGIP 2022); 127051E (2023) https://doi.org/10.1117/12.2680442
Event: Fourteenth International Conference on Graphics and Image Processing (ICGIP 2022), 2022, Nanjing, China

Abstract

In recent years, to ensure the stable development of the marine economy, the analysis and processing of maritime vessel targets combined with natural light images have played a significant role. However, in scenes such as ports, there are sometimes background noises similar to vessel features, so the classifier based on the convolutional neural network cannot achieve good classification results. To pay more attention to effective information when classifying natural maritime images, we propose an efficient channel attention module (ECAM) and a re-parameterized spatial attention module (RSAM) through one-dimensional convolution and re-parameterization method. This paper combines ECAM and RSAM to present an efficient re-parameterized convolutional block attention module (ER-CBAM) to classify natural maritime images. Besides, in response to the current lack of large-scale natural marine image datasets, we established a natural maritime vessel image (NMVI) dataset. Experiments on NMVI show that by combining the proposed attention module, ResNet50 can achieve a top-1 accuracy score of 85.32% with almost no extra computational consumption and a 3.02% improvement compared to the previous 82.30%, which suggests that the proposed method is suitable for maritime scenarios.

Citation Download Citation

Shi-Jia Zhao, Jia-Chun Zheng, Lei Zhang, and Yan-duo Hu "Vessel classification algorithm based on the convolutional attention module for natural maritime images", Proc. SPIE 12705, Fourteenth International Conference on Graphics and Image Processing (ICGIP 2022), 127051E (27 June 2023); https://doi.org/10.1117/12.2680442

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

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.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available