本发明属于图像处理与立体视觉技术领域，具体涉及一种基于深度卷积网络的可见光舰船虚警剔除方法。
背景技术：
利用遥感图像对广域海洋内的舰船进行检测，在民用领域内，可为海洋渔业管理提供及时的数据支撑；在军用领域内，可以快速的掌握争端海域内舰船的态势，为军事决策提供高时效的信息。
传统的舰船识别方法主要为基于底层特征结合传统机器学习分类器的方法。该方法一般需要人为构造有效的底层特征，如颜色特征：颜色直方图、颜色矩、颜色相关图等，形状特征：hog、sift、bovw等，纹理特征：lbp、灰度共生矩阵等，然后将提取得到的特征送入分类器(如贝叶斯分类器、支持向量机等)进行分类识别。此类技术需要开发人员具备深入的领域知识，并在此基础上针对性地开发适用于任务的特定描述特征，其开发难度大，收效不甚理想。考虑到对于舰船检测虚警剔除领域内的特殊工程问题，目标特性易受到运动尾迹与侧迹的影响，在目标的技术条件下，未能够彻底的去除目标运动产生尾迹侧迹对目标特性的干扰，其鉴别能力弱，虚警多。
技术实现要素：
为了解决上述技术问题，本发明提出一种基于深度卷积网络的可见光舰船虚警剔除方法，该方法能够更准确地区分图像中的舰船和虚警，从而解决舰船检测中虚警率特别高的问题。
实施本发明的技术方案如下：
一种基于深度卷积网络的可见光舰船虚警剔除方法，具体步骤为：
步骤1、选取光学遥感图像中的舰船切片和虚警切片，作为初始的舰船样本和虚警样本；对两类样本中的切片进行舰船、虚警类型标记；
步骤2、分别对初始的舰船样本和虚警样本进行数据增广；
步骤3、分别对增广后的舰船样本和虚警样本进行传统特征提取，生成舰船样本集特征库与虚警样本集特征库，使用k-means算法结合样本集特征库将增广后的舰船样本细分为m个子类、将增广后的虚警样本分为n个子类，共得到(m+n)个子类样本，对(m+n)个子类样本中的切片进行类型标记；
步骤4、使用已经在imagenet大数据集上训练过的vgg19网络作为基网络，并在此基础上进行优化，得到优化后的cnn卷积神经网络；
步骤5、初始化优化后的cnn中的待训练参数，并设置网络超参数，得到训练网络；
步骤6、将步骤3得到的(m+n)个子类样本打乱，并使用训练网络对其进行迁移学习训练，共训练t1代，训练过程中每一代都会自动更新待训练参数，并获得对应的训练精度和损失曲线；
步骤7、根据步骤6中获得的训练精度和损失函数曲线选择最优的那一代训练网络中的待训练参数，加载到步骤4中得到的cnn构成判定网络，对打乱后的(m+n)个子类样本进行类型判定，记录样本中每个切片的判定概率；
步骤8、将打乱后的(m+n)个子类样本进行重新归类，然后根据判定概率将(m+n)个子类样本中的切片各自细分成k小类，最终得到k×(m+n)个小类，分类后对切片进行类型标记；
步骤9、根据k×(m+n)个小类中每小类的数据量对k×(m+n)个小类分别进行数据增广，得到k×(m+n)个训练样本集；
步骤10、将在步骤9中得到的k×(m+n)个训练样本集打乱，“冻结”步骤5中得到的训练网络的卷积层，然后对打乱后的训练样本集进行微调训练，即只调节全连接层的待训练参数，不调节卷积层的待训练参数；训练t2代后结束，训练过程中每一代都会自动更新待训练参数，并获得对应的训练精度和损失曲线；
步骤11、根据步骤10中获得的训练精度和损失函数曲线选择最优的那一代训练网络中的待训练参数，加载到步骤4中得到的cnn构成最终识别网络。
进一步的，步骤2和步骤9中所述数据增广的方式有仿射变换、亮度变换和对比度变换。
进一步的，步骤3具体为：
1)对切片进行传统特征提取包含目标长度、长宽比、面积、空间扩展度、占空比五个基础特征；
2)采用k-means算法结合1)所提出的5个基础特征对舰船切片与虚警切片分别进行子类划分，将舰船切片细分为m类子类，将虚警切片细分为n个子类。
进一步的，步骤4所述优化方式为：
在基网络前两个最大池化层的前面和每一个卷积层的后面分别加入batchnorm策略。
进一步的，步骤7所述判定概率为：对样本进行类型判定时，样本中每个切片都会获得(m+n)个概率，分别对应不同的类型，(m+n)个概率的概率和为1，将(m+n)个概率中最大的那个概率记为判定概率。
进一步的，步骤8所述分类方式为：
以“判定概率在0.85以上”、“判定概率在0.75～0.85”和“判定概率在0.75以下”为分类区间，将(m+n)个子类样本中的切片各自细分成3小类，即k＝3，最终得到3×(m+n)个小类，根据判定概率判定切片的类型，如果切片类型判定错误，将该切片归为对应子类的“判定概率在0.75以下”小类里。
进一步的，步骤9所述根据小类的数据量对小类进行数据增广的具体方式如下：
(m+n)个子类中的每个子类都有3个小类，针对每个子类，数据量最大的小类不用进行数据增广；对其余两小类进行数据增广，使这两个小类各自的数据量增广至数据量最大的小类的数据量的2/3。
有效益处
本发明提供的一种基于深度卷积网络的可见光舰船虚警剔除方法，和现有技术相比，其优点在于：
1)本发明针对舰船与虚警两类识别问题，考虑到舰船类内特征差异大对识别网络准确性的影响，使用传统的特征提取与k-means聚类技术相结合的方法将相同的特征的舰船与虚警进行初步细分类，使得对网络训练时网络针对性更强，鉴别能力更强；
2)本发明针对舰船检测中难分虚警的问题，用训练网络本身对所有样本进行类型判定，以判定概率区间为标准对样本进行细分后再训练，有效提高了区分虚警和舰船的准确率，进而提高了整个网络针对具体场景的识别准确率。
附图说明
图1为本发明流程图；
图2为舰船与虚警细分类示意图；
其中舰船目标分为大船(a)(b)与小船两个(c)(d)子类，虚警细分为五个子类，示意图(e)(f)(g)(h)(c)。
具体实施方式
下面结合附图并举实施例，对本发明进行详细描述，应当理解，此处所描述的优选实施例仅用于说明和解释本发明，并不用于限定本发明。
实施例1：
针对舰船检测中虚警率特别高的问题，本发明提供一种基于深度卷积网络的可见光舰船虚警剔除方法，该方法能更好地区分切片中的舰船和虚警，根据图1的流程图，以5米空间分辨率光学遥感图像数据集作为实验对象进行实验，具体步骤为：
步骤1、采用5米可见光256像素*256像素大小的光学遥感图像为数据集，选取光学遥感图像中的舰船切片，作为初始的舰船样本；选取光学遥感图像中的虚警切片，作为初始的虚警样本，对两类样本中的切片进行舰船、虚警类型标记，初始时每类样本有100幅切片；
步骤2、利用旋转、平移、微小扭曲等仿射变换及亮度、对比度变换分别对初始的舰船样本和虚警样本进行数据增广，使每类样本的数据量增广1倍；
步骤3、分别对增广后的舰船样本和虚警样本进行传统特征提取，生成舰船样本集特征库与虚警样本集特征库，提取的特征包括目标长度、长宽比、面积、空间扩展度、占空比五个基础特征，使用k-means算法结合舰船样本集特征库将增广后的舰船样本细分为2个子类，使用k-means算法结合虚警样本集特征库将增广后的虚警样本分为5个子类，共得到7个子类样本，如图2所示，对7个子类样本中的切片进行类型标记；
步骤4、使用已经在imagenet大数据集上训练过的vgg19网络作为基网络，并在此基础上进行优化，得到优化后的cnn(convolutionalneuralnetwork，卷积神经网络)；即在基网络前两个最大池化层的前面和每一个卷积层的后面分别加入batchnorm策略，即对训练的每一批数据{x1,x2,…,xn}有：
其中，
步骤5、初始化优化后的cnn中的待训练参数，并将基网络的网络超参数的数值赋给优化后的cnn中的网络超参数，然后将网络学习率修改为0.0005，将每批训练的数据量n修改为20，得到训练网络；
步骤6、将步骤3得到的7个子类样本打乱，并使用训练网络对其进行迁移学习训练，共训练10代，训练过程中每一代都会自动更新待训练参数，并获得对应的训练精度和损失曲线；
步骤7、根据步骤6获得的训练精度和损失函数曲线选择最优的那一代训练网络中的待训练参数，加载到步骤4中得到的优化后的cnn构成判定网络，对打乱后7个子类样本进行类型判定，记录样本中每个切片被判定为不同类型的概率，即针对每个切片都会获得7个概率，分别对应不同的类型，7个概率的概率和为1，将7个概率中最大的那个概率记为判定概率；
步骤8、将打乱后的7个子类样本进行重新归类，然后以“判定概率在0.85以上”、“判定概率在0.75～0.85”和“判定概率在0.75以下”为分类区间将7个子类样本中的切片分别细分成3小类，最终得到21个小类；根据判定概率判定切片的类型，如果切片类别判定错误，将该切片归为对应子类的“判定概率在0.75以下”小类的类别，分类后对切片进行类型标记；
步骤9、利用旋转、平移、微小扭曲等仿射变换及亮度、对比度变换，并根据21个小类中每小类的数据量对21个小类分别进行数据增广，得到21个训练样本集；
根据小类数据量对小类进行数据增广的具体方式如下：7个子类中的每个子类都有3个小类，针对每个子类，“识别概率在0.85以上”小类的数据量最大，不用进行数据增广；对其余两小类进行数据增广，使这两个小类的数据量增广至“识别概率在0.85以上”小类的数据量的2/3；
步骤10、将在步骤9中得到的21个训练样本集打乱，“冻结”步骤5中得到的训练网络的卷积层，然后对打乱后的训练样本集进行微调训练，即只调节全连接层的待训练参数，不调节卷积层的待训练参数；训练5代后结束，训练过程中每一代都会自动更新待训练参数，并获得对应的训练精度和损失曲线；
步骤11、根据步骤10获得的训练精度和损失函数曲线选择最优的那一代训练网络中的待训练参数，加载到步骤4中得到的优化后的cnn构成最终识别网络。
以3/10的比例随机从打乱的训练样本集中抽出一部分样本构成验证样本集，用最终识别网络对其进行识别验证，将小类识别结果归为相对应的舰船和虚警两大类，输出最终识别结果，经验证，该方法将舰船检测的准确率提高了6％。
以上所述仅为本发明的优选实施例而已，并不用于限制本发明，尽管参照前述实施例对本发明进行了详细的说明，对于本领域的技术人员来说，其依然可以对前述各实施例所记载的技术方案进行修改，或者对其中部分技术特征进行等同替换。凡在本发明的精神和原则之内，所作的任何修改、等同替换、改进等，均应包含在本发明的保护范围之内。