Ship classification for unbalanced SAR dataset based on convolutional neural network

Jianwei Li; Changwen Qu; Shujuan Peng

doi:10.1117/1.JRS.12.035010

6 August 2018 Ship classification for unbalanced SAR dataset based on convolutional neural network

Jianwei Li, Changwen Qu, Shujuan Peng

Author Affiliations +

Journal of Applied Remote Sensing, Vol. 12, Issue 3, 035010 (August 2018). https://doi.org/10.1117/1.JRS.12.035010

Abstract

Ship classification in synthetic aperture radar (SAR) images is essential in remote sensing but still full of challenges in the deep learning era. The unbalanced dataset and lack of models are two limitations. Upsampling with data augmentation and ratio batching are proposed to solve the first problem. Upsampling with data augmentation is upsampling by cropping and flipping. It can improve the diversity of the dataset. Ratio batching is realized by choosing the same amount of ships per class in each minibatch. It can make the model converge faster and better. To solve the second problem, a new loss function and convolutional neural network model are proposed. The new loss function can maximize the intraclass compactness and interclass separation simultaneously. Dense residual network has two submodules. One is the identity mapping through elementwise summation to reuse old features. The other is dense connection through concatenation to exploit new features. The designed architecture is suitable for the task of SAR ship classification. We use the confusion matrix and accuracy averaged on classes to measure the performance. From the experiments, we can find that the proposed ideas have excellent performance especially on the rare classes.

Citation Download Citation

Jianwei Li, Changwen Qu, and Shujuan Peng "Ship classification for unbalanced SAR dataset based on convolutional neural network," Journal of Applied Remote Sensing 12(3), 035010 (6 August 2018). https://doi.org/10.1117/1.JRS.12.035010

Received: 30 March 2018; Accepted: 10 July 2018; Published: 6 August 2018

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