Lightweight Mask RCNN for Warship Detection and Segmentation

Abstract

As the term X(Everything)+AI indicates, AI is applied in every aspect of current societies. Likewise, the military requirements for AI are increasing as well. AIs that automatically detect and classify objects are required for surveillance and reconnaissance. Especially in terms of naval operation, identifying types of warships and recognizing mounted armaments have significance as the first step of the operation. This study is the proposal of an AI model that can identify warships’ type and weapon by analyzing video information taken on sea, and evaluate threat priority and response level. The proposed model is based on Mask RCNN, the Image Segmentation model, but was designed in a lightweight version, so that it could be used on a platform of the vessel where the use of high performing computers is limited. To lightweight the model, the former backbone was replaced with MobileNetV2, and the convolution operation of the RPN was replaced with Depthwise Separable Convolution operation, which operates respectively in each channel. The lightweight Mask RCNN model showed 64% lower number of parameters compared to the base model. However, its mAP, the classification accuracy, was similar with the base model.

Keywords

• Warship detection and segmentation
• lightweight deep learning
• Mask RCNN
• MobileNet