Ship Classification in Remote Sensing Images using FastAI

Abstract

Specifying ship categories in waterways plays an important role in the field of marine surveillance, especially when classification is performed from satellite images due to the advancement in remote sensing technologies. In this paper, we presented an approach for ship classification of optical remote sensing images. Our approach was based on two aspects, modifying models and applying additional techniques to improve accuracy of classification. Two pretrained models, MobileNetV2 and DenseNet121, were modified in this work and all techniques were implemented using Fastai library. To illustrate the effectiveness of our approach, we compared the accuracy of the modified models to the original one. A public Dataset for Ship Classification in Remote sensing images (DSCR), containing six military ship types and a civilian ship type, was used for evaluation. The results showed that our modified DenseNet121 achieved the best accuracy at 99.52% and also outperformed the benchmark result of ResNet101 reported from the original dataset.