[发明专利]一种基于边缘主体融合信息的遥感目标显著性检测方法

说明书

本发明公开了一种基于边缘主体融合信息的遥感目标显著性检测方法。本发明步骤：1、图像数据预处理；2、网络搭建：网络包括encoder部分和decoder部分；encoder部分包括一个输入卷积层和六个卷积块，其中前四个卷积块采用ResNet34；encoder和decoder之间添加3个3*3的卷积层作为桥接；Decoder部分与encoder部分为对称结构，每个解码块与对应encoder部分concat的同时，还将前一个解码块concat到当前解码块；3、数据输入和训练。本发明有效利用了编码阶段的边缘信息和主体信息，并选择使用多loss融合的模式对U‑net解码部分的各个边缘输出进行约束，同时对各个边缘输出进行融合，使得遥感目标的检测取得了显著提升。

技术领域

本发明属于计算机视觉领域，涉及显著目标检测、遥感图像检测领域。具体涉及一种基于边缘主体融合信息的遥感目标显著性检测方法。

背景技术

随着深度学习、神经网络的飞速发展，计算机视觉领域实现了前所未有的跨越。目标检测作为计算机视觉领域一个经典大类，受到广泛的研究和探讨，在显著目标检测、行人重识别、图像数据评估等各个方向都取得了巨大进步。在生活中，面部扫描、车牌扫描、天网工程等都利用到了目标检测的相关技术。

人类视觉系统在面对自然场景时具有快速搜索和定位感兴趣目标的能力，这种视觉注意机制是人们日常生活中处理视觉信息的重要机制。随着互联网带来的大数据量的传播，如何从海量的图像和视频数据中快速地获取重要信息，已经成为计算机视觉领域一个关键的问题。通过在计算机视觉任务中引入这种视觉注意机制，即视觉显著性，可以为视觉信息处理任务带来一系列重大的帮助和改善。显著性物体检测的目的即是从图像中定位最有吸引力和视觉上独特的物体或区域，大量应用于图像分割、目标重定位、目标图像前景注释等邻域。遥感图像的显著目标检测在检测上又不同于常规显著性检测任务，遥感图像通过航空摄影、航空扫描、微波雷达等手段，以飞机、轮船等交通工具，公路、河流、运动场馆等目标物体为主要分割对象，目标在图像中往往较小且复杂。遥感图像的显著目标检测在自然保护、雷达检测等邻域有巨大作用。

U-Net是现在最为流行的图像分割网络，网络主要包括下采样和上采样两个部分，通过下采样对图片特征进行提取处理，而上采样则可以将特征信息进行还原，使得网络最后输出为一个完整图像。

随着深度神经网络的飞速发展，遥感图像的显著目标检测在检测精度以及检测速度上都有巨大进步，人们也发现在目标检测中加入边缘信息后能够使得检测结果有进一步提升。但是现存的方法只是单纯地将边缘信息加入到主体检测中，几乎没人关注到主体检测和边缘检测的协同优化效果。本文有效利用了边缘信息和主体信息，利用边缘信息指导主体，同时利用主题信息反馈指导边缘，使得显著目标检测，尤其是遥感目标的显著目标检测结果有了巨大提升。

发明内容

针对现有技术存在的问题，本发明提供一种基于边缘主体融合信息的遥感目标显著性检测方法。

本发明解决其技术问题所采用的技术方案包括如下步骤：

步骤1、图像数据预处理，获得预处理后的图像训练集：

训练数据集采用ORSSD图像数据集。对待训练数据集进行图像预处理，首先去除图像相关噪声干扰使数据更加精确，并利用matlab工具获得只含边缘轮廓的图像训练集，然后对获取的图像训练集进行扩增，从而获得更好的训练效果。

步骤2、网络搭建：

网络主要结构如图1，本方法采用encoder-decoder办法，网络结构主体参考U-net，具体的：U-net的encoder部分包括一个输入卷积层和六个卷积块，其中前四个卷积块采用ResNet34，为了得到和传统ResNet34相同的感受野，在四个卷积块后又添加了两个卷积块。为了更好的获得全局信息，在网络的encoder和decoder之间添加3个3*3的卷积层作为桥接。