Ship Classification in High-Resolution SAR Images Using Deep Learning of Small Datasets
AbstractWith the capability to automatically learn discriminative features, deep learning has experienced great success in natural images but has rarely been explored for ship classification in high-resolution SAR images due to the training bottleneck caused by the small datasets. In this paper, convolutional neural networks (CNNs) are applied to ship classification by using SAR images with the small datasets. First, ship chips are constructed from high-resolution SAR images and split into training and validation datasets. Second, a ship classification model is constructed based on very deep convolutional networks (VGG). Then, VGG is pretrained via ImageNet, and fine tuning is utilized to train our model. Six scenes of COSMO-SkyMed images are used to evaluate our proposed model with regard to the classification accuracy. The experimental results reveal that (1) our proposed ship classification model trained by fine tuning achieves more than 95% average classification accuracy, even with 5-cross validation; (2) compared with other models, the ship classification model based on VGG16 achieves at least 2% higher accuracies for classification. These experimental results reveal the effectiveness of our proposed method. View Full-Text
Share & Cite This Article
Wang, Y.; Wang, C.; Zhang, H. Ship Classification in High-Resolution SAR Images Using Deep Learning of Small Datasets. Sensors 2018, 18, 2929.
Wang Y, Wang C, Zhang H. Ship Classification in High-Resolution SAR Images Using Deep Learning of Small Datasets. Sensors. 2018; 18(9):2929.Chicago/Turabian Style
Wang, Yuanyuan; Wang, Chao; Zhang, Hong. 2018. "Ship Classification in High-Resolution SAR Images Using Deep Learning of Small Datasets." Sensors 18, no. 9: 2929.
Note that from the first issue of 2016, MDPI journals use article numbers instead of page numbers. See further details here.