摘要
一种基于残差网络和U-Net分割网络的卫星图像分割方法,其步骤为:构建残差网络ResNet34;构建U-Net分割网络;构建训练样本集;训练残差网络ResNet34;训练U-Net分割网络;将待分割的卫星图像输入到残差网络ResNet34进行二分类,判断包含船只目标;使用U-Net分割网络对分类结果中的正样本进行二值分割;对于分类结果中的负样本,直接输出单值掩码图;本发明利用残差网络ResNet34对卫星图像进行二分类,使用U-Net分割网络仅对分类结果中的正样本进行分割,并在U-Net分割中嵌入SE-ResNet模块,提取更加精细的分割掩码,实时性高,分割精度高。
- 单位