Ship Classification Based on Improved Convolutional Neural Network Architecture for Intelligent Transport Systems

被引:15
|
作者
Leonidas, Lilian Asimwe [1 ]
Jie, Yang [1 ]
机构
[1] Wuhan Univ Technol, Sch Informat Engn, Wuhan 430081, Peoples R China
基金
美国国家科学基金会;
关键词
convolutional neural network; inland waterways; deep learning;
D O I
10.3390/info12080302
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
In recent years, deep learning has been used in various applications including the classification of ship targets in inland waterways for enhancing intelligent transport systems. Various researchers introduced different classification algorithms, but they still face the problems of low accuracy and misclassification of other target objects. Hence, there is still a need to do more research on solving the above problems to prevent collisions in inland waterways. In this paper, we introduce a new convolutional neural network classification algorithm capable of classifying five classes of ships, including cargo, military, carrier, cruise and tanker ships, in inland waterways. The game of deep learning ship dataset, which is a public dataset originating from Kaggle, has been used for all experiments. Initially, the five pretrained models (which are AlexNet, VGG, Inception V3 ResNet and GoogleNet) were used on the dataset in order to select the best model based on its performance. Resnet-152 achieved the best model with an accuracy of 90.56%, and AlexNet achieved a lower accuracy of 63.42%. Furthermore, Resnet-152 was improved by adding a classification block which contained two fully connected layers, followed by ReLu for learning new characteristics of our training dataset and a dropout layer to resolve the problem of a diminishing gradient. For generalization, our proposed method was also tested on the MARVEL dataset, which consists of more than 10,000 images and 26 categories of ships. Furthermore, the proposed algorithm was compared with existing algorithms and obtained high performance compared with the others, with an accuracy of 95.8%, precision of 95.83%, recall of 95.80%, specificity of 95.07% and F1 score of 95.81%.
引用
收藏
页数:13
相关论文
共 50 条
  • [1] An Improved Convolutional Neural Network Architecture for Image Classification
    Ferreyra-Ramirez, A.
    Aviles-Cruz, C.
    Rodriguez-Martinez, E.
    Villegas-Cortez, J.
    Zuniga-Lopez, A.
    PATTERN RECOGNITION, MCPR 2019, 2019, 11524 : 89 - 101
  • [2] An Intelligent Ship Image/Video Detection and Classification Method with Improved Regressive Deep Convolutional Neural Network
    Huang Zhijian
    Sui Bowen
    Wen Jiayi
    Jiang Guohe
    COMPLEXITY, 2020, 2020
  • [3] Deep Convolutional Neural Network based Ship Images Classification
    Mishra, Narendra Kumar
    Kumar, Ashok
    Choudhury, Kishor
    DEFENCE SCIENCE JOURNAL, 2021, 71 (02) : 200 - 208
  • [4] Ship Classification Based on Multifeature Ensemble with Convolutional Neural Network
    Shi, Qiaoqiao
    Li, Wei
    Tao, Ran
    Sun, Xu
    Gao, Lianru
    REMOTE SENSING, 2019, 11 (04)
  • [5] Intelligent waste classification approach based on improved multi-layered convolutional neural network
    Chhabra M.
    Sharan B.
    Elbarachi M.
    Kumar M.
    Multimedia Tools and Applications, 2024, 83 (36) : 84095 - 84120
  • [6] Ship classification for unbalanced SAR dataset based on convolutional neural network
    Li, Jianwei
    Qu, Changwen
    Peng, Shujuan
    JOURNAL OF APPLIED REMOTE SENSING, 2018, 12 (03)
  • [7] Research on Ship Trajectory Classification Based on a Deep Convolutional Neural Network
    Guo, Tao
    Xie, Lei
    JOURNAL OF MARINE SCIENCE AND ENGINEERING, 2022, 10 (05)
  • [8] Classification of Electronic Components Based on Convolutional Neural Network Architecture
    Atik, Ipek
    ENERGIES, 2022, 15 (07)
  • [9] TextConvoNet: a convolutional neural network based architecture for text classification
    Sanskar Soni
    Satyendra Singh Chouhan
    Santosh Singh Rathore
    Applied Intelligence, 2023, 53 : 14249 - 14268
  • [10] TextConvoNet: a convolutional neural network based architecture for text classification
    Soni, Sanskar
    Chouhan, Satyendra Singh
    Rathore, Santosh Singh
    APPLIED INTELLIGENCE, 2023, 53 (11) : 14249 - 14268