Ship target detection is of great significance in marine surveillance, rescue and so on. In this paper, in order to improve the performance of ship target detection, we proposed a ship target detection method based on multi task learning. There are mainly two contributions. Firstly, we designed a multi-task learning model by integrating segmentation module to the faster RCNN model. Through the strategies of feature sharing and joint learning, it is helpful to improve the accuracy of target detection with the assistance of segmentation; Secondly, in order to deal with the impact of initial anchor frame scale on target detection accuracy, we introduced an adaptive anchor width height ratio setting method based on improved K-means algorithm, by adaptively select initial anchor size suitable for the characteristics of ship targets, it is beneficial to further improve the detection accuracy. Moreover, we constructed an extended version of ship image data set including 14614 images belonging to 13 categories. Experimental results demonstrated that the proposed model can effectively improve the accuracy of ship target detection; and the comparison and the ablation experiments further validated the strategies of multi-task joint learning and adaptive anchor size setting is helpful for improving the performance of ship target detection.
We propose a novel LBCNN model with AM Softmax based on bilinear CNN (BCNN) and AM Softmax loss function, which can better fit fine-grained birds recognition tasks. There are mainly two contributions. Firstly, in order to reduce the model size and recognition time, we design a lightweight BCNN model to reduce the parameters. We replace original VGG16 backbone with MobileNet structure which decomposes the convolution operation into two smaller operations: depthwise revolution and pointwise revolution. Secondly, to make up for the decrease in accuracy, we introduce the Additive Margin Softmax (AM Softmax) loss function to enhance the discrimination ability. By comprehensive discussion of the influence of different parameter settings and different loss functions, we test the proposed lightweight BCNN on the bird dataset CUB-200-2011. Experimental results demonstrate that the proposed model can achieve comparable results with much fewer parameters.
Access to the requested content is limited to institutions that have purchased or subscribe to SPIE eBooks.
You are receiving this notice because your organization may not have SPIE eBooks access.*
*Shibboleth/Open Athens users─please
sign in
to access your institution's subscriptions.
To obtain this item, you may purchase the complete book in print or electronic format on
SPIE.org.
INSTITUTIONAL Select your institution to access the SPIE Digital Library.
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.