Laser & Optoelectronics Progress, Volume. 60, Issue 6, 0610005(2023)

Application of a Lightweight Convolutional Neural Network in Ship Classification

Wenliang Wang, Xiaodi Yang*, Boya Zhang, Jishun Ma, Peng Zeng, and Peng Han
Author Affiliations
  • CSSC (Zhejiang) Ocean Technology Co., Ltd., Zhoushan 316000, Zhejiang, China
  • show less

    A lightweight convolutional neural network, AGNet-improved GhostNet-50, is proposed for ship classification using self-made ship datasets while ensuring a desirable classification accuracy with a small model size. First, a Ghost module that integrates asymmetric convolution is proposed to improve the feature extraction capability of the AGNet convolution process. Next, combined with the bottleneck structure, an asymmetric Ghost bottleneck module is designed, further reducing the computational cost while maintaining the expression ability of the model. Finally, a 1×1 convolution layer in GhostNet-50 is removed to reduce the parameter redundancy of the overall model. The proposed method was compared from multiple aspects using evaluation indexes such as classification accuracy, parameters, and computational and inference speeds. Based on the experimental results, the accuracy of the AGNet model in the test set of 33 categories reaches 93.87%, and the number of model parameters is only 0.72×106. Compared with GhostNet-50, the AGNet model is compressed in terms of size by 46.67%, and the accuracy is improved by 2.93 percentage points. The experimental results show that AGNet can achieve better classification performance with a smaller model size and can be better applied to ship classification tasks.

    Tools

    Get Citation

    Copy Citation Text

    Wenliang Wang, Xiaodi Yang, Boya Zhang, Jishun Ma, Peng Zeng, Peng Han. Application of a Lightweight Convolutional Neural Network in Ship Classification[J]. Laser & Optoelectronics Progress, 2023, 60(6): 0610005

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Image Processing

    Received: Nov. 23, 2021

    Accepted: Jan. 17, 2022

    Published Online: Mar. 7, 2023

    The Author Email: Yang Xiaodi (1755018902@qq.com)

    DOI:10.3788/LOP213033

    Topics