基于MaskR-CNN的卫星影像船舶检测研究赵亮马旺叶杨雯彦曹媛媛【摘要】本文提出了一种基于MaskR-CNN实现船舶检测的方法，通过在网络训练过程中随机采样目标像素到背景中的其他位置，改善了目标与背景不平衡的问题。对比分析基础模型与本文方案，检测与分割的精度都有了显著的提升。【关键词】MaskR-CNN;船舶;检测;分割：O436：A：2095-2457（2019）30-0024-002DOI：10.19694/j.cnki.issn2095-2457.2019.30.011SatelliteImageShipDetectionBasedOnMaskR-CNNZHAOLiangMAWang-yeYANGWen-yanCAOYuan-yuan（SchoolofGeomaticsandMarineInformation，LianyungangJiangsu222005，China）【Abstract】Inthispaper，amethodbasedonMaskR-CNNwasproposedtorealizeshipdetection.Byrandomlysamplingthetargetpixelstootherpositionsinthebackgroundduringnetworktraining，theimbalancebetweenthetargetandthebackgroundwasimproved.Bycomparingandanalyzingthebasicmodelandtheschemeinthispaper，theaccuracyofdetectionandsegmentationhasbeensignificantlyimproved.【Keywords】MaskR-CNN;Ship;Detection;Segmentation0引言海运承担着国际贸易进出口的重要任务，根据卫星影像获得船舶的位置与大小具有重要的军事与商业作用。卫星影像船舶检测主要有以下两个难点：（1）样本不均衡的问题，通常数据集中的船舶数量与像素占比较少，样本不均衡使得检测模型更关注较大的目标，忽略较小的船舶，容易造成漏检与误检。（2）由于目标覆盖的区域较小，造成船舶的位置缺少多样性，模型在验证通用性时变得困难。当前深度学习检测算法利用图像的多层特征优化对小目标检测的精度，主要有：Tsung-YiLin等[1]提出的图像金字塔网络（FPN），通过对训练图片上采样出多尺度的图像金字塔，增强小目标的细粒度特征，优化定位和识别效果。第二种方法为逐层预测，即对卷积神经网络抽取的每层特征图输出进行一次预测，综合考量得出结果[2]。以上两种方法对于硬件算力有极高的要求，且不是专门针对船舶检测而设计的，模型的鲁棒性较差。本文提出的方法，在模型训练过程中将船舶目标特征图随机采样复制到同批次中的其他图像背景中，改善不平衡问题的同时，模型需要的计算量也没有增加。实验证明，即使像素占比非常小的船舶，检测与分割的精度也得到了有效的提升。1算法模型本文方法基于MaskR-CNN目标检测框架修改而来，MaskR-CNN是在FasterR-CNN的基础上增强了一个平行的smallFCN（全卷积网络）分支，以获得目标的轮廓定位，从而实现准确的分割。因此基于本文方法，可以同时获得船舶目标检测与实例分割结果。图1smallFCN结构图为了适应船舶检测的需要，本文对模型做了针对性的改进。首先，预处理数据，采用传统的语义分割模型，获得海水与船舶的大致边界信息并传入训练网络;其次对船舶标注图像聚类分析获取船舶大小的分布数据，确定了检测过程不同大小检测框的数量;然后在网络训练过程中，上采样船舶特征矩阵，对抗网络（GAN）生成船舶的相似特征图，最后基于边界信息复制到同批次训练数据的背景中，确保不会产生船舶重叠。本文方法同样考虑到不同的IoU（交并比）阈值对于样本的选取产生的影响，使用了渐进IOU的方法，在特征金字塔的每一层添加多个检测头，逐渐提高IOU的阈值，确保不会漏掉可能的船舶目标。2实验结果与分析实验数据选自Kaggle提供的28G卫星遥感船舶影像数据，共5000张，按照4：1的比例划分为训练集于测试集，在GoogleClob平台上利用Tensnorflow深度学习框架实现，训练了25批次后，网络的损失趋于稳定，部分检测结果如下图所示：图2检测与分割成果为了测试本文方法的精度，从原始数据中重新随机抽取了200个目标测试数据。分别基于未改进的MaskR-CNN（文献[1]方法）与本文方法进行测试。目标检测常用准确率（P）、召回率（R）评价模型的精度，计算公式为：P=TP/（TP+FP）R=TP/（TP+FN）=TP/P其中，TP：被正确地划分为船舶的实例个数;FP：被错误地划分为船舶的个数;FN：被错误地划分为海水背景的个数。同时计算分割结果与原始标注的像素相交面积比，得到分割的精度AP。计算结果如表1所示。从表1中可以看...