计算机工程与应用2019，Vol.55Issue(22)：146-151,6.DOI:10.3778/j.issn.1002-8331.1902-0144
遥感图像舰船检测的旋转卷积集成YOLOv3模型
Rotation Convolution Ensemble YOLOv3 Model for Ship Detection in Remote Sensing Images
摘要
Abstract
The orientation diversity of target in remote sensing images caused by the bird-eye view affects the rotation invariance of ship detection with large aspect ratio. Aiming at this problem, this paper proposes an incline bounding box detection model based on improved YOLOv3. The incline bounding box regression is implemented by introducing angle prediction. A rotation convolution ensemble module is proposed to increase the awareness for Deep Convolutional Neural Networks(DCNN)feature map to angle change by rotation convolution and rotation activation. The bounding box angle prediction is modeled as a two-step classification problem from coarse to fine. The error of angle classification punish-ment is introduced into the multi-task loss function of the model, so that the model can learn the angular offset of the object. Experiments on ship dataset show that the proposed model increases the Average Precision(AP)of ship detection in the test dataset by 12.7% compared with the classic YOLOv3 model, while maintaining the advantage of single-stage model in detection speed.关键词
遥感图像/目标检测/舰船检测/旋转卷积/深度学习Key words
remote sensing images/object detection/ship detection/rotation convolution/deep learning分类
信息技术与安全科学引用本文复制引用
吴止锾,李磊,高永明..遥感图像舰船检测的旋转卷积集成YOLOv3模型[J].计算机工程与应用,2019,55(22):146-151,6.基金项目
军内科研项目. （）
