[发明专利]一种基于金字塔输入增益的卷积神经网络的目标检测方法

摘要

本发明涉及一种基于金字塔输入增益的卷积神经网络的目标检测方法，属于计算机视觉及目标检测技术领域。所述目标检测方法基于包括特征提取模块和多任务预测模块的卷积神经网络模型PiaNet；所述目标检测方法，包括训练阶段和测试阶段；训练阶段采用两阶段迁移学习策略，包括：步骤(1)数据增强和数据预处理，产生第一阶段训练的训练集、第二阶段训练的训练集和测试集；步骤(2)在二分类网络中进行第一阶段训练；步骤(3)进行第二阶段训练，得到训练后的PiaNet网络；测试阶段即精确检测目标，具体为：将测试集输入到训练后的PiaNet网络，通过多任务损失函数输出检测框位置和分类结果。适用面广泛，且具有很高的鲁棒性。

主权项

1.一种基于金字塔输入增益的卷积神经网络的目标检测方法，其特征在于：提出了一种基于金字塔输入增益的卷积神经网络模型PiaNet，网络模型PiaNet主要包括特征提取模块和多任务预测模块；其中，所述特征提取模块包括特征压缩模块和特征放大模块；其中，特征压缩模块包括N个卷积块，沿着一条融合源连接输入的特征压缩路径展开；特征放大模块包括M个反卷积块，沿着一条融合了跳跃连接的特征放大路径展开；其中，源连接对预处理后的图像分别进行average pooling操作得到多尺度源图像并构成了图像金字塔，降低了特征压缩路径上的信息损失；特征压缩路径上的每个卷积块包含一次卷积、BN、激活和Pooling操作，对输入图像进行特征提取得到特征图；从特征压缩路径上的第一个卷积块开始，每一层输出的特征图和相应尺度的源图像拼接构成下一层卷积操作的输入；特征放大路径上的每个卷积块包含一次Unpooling、反卷积、BN和激活操作，对特征压缩路径输出的特征图进行特征放大；从特征放大路径上的第一个反卷积块开始，每一层输出的特征图和来自特征压缩路径的特征图拼接构成下一层反卷积操作的输入，使得压缩路径上具有的目标细节信息能够传输到具有高层语义信息的放大特征层路径上；多任务预测模块包含多任务损失函数；所述目标检测方法，包括训练阶段和测试阶段；其中，训练阶段采用两阶段迁移学习策略，具体实现步骤如下：步骤(1)数据增强和数据预处理，具体为：基于数据划分、数据增强以及预处理从带标记的原始数据中产生第一阶段训练的训练集、第二阶段训练的训练集和测试集；其中，数据增强的使用是为了解决了小样本的学习问题；步骤(2)基于步骤(1)生成的第一阶段训练所需的训练集进行第一阶段训练；其中，第一阶段训练在二分类网络中进行，二分类网络由PiaNet模型中的特征提取模块和average pooling层以及softmax层相连接组成；步骤(2)的具体训练过程，具体为：将步骤(1C)获得的用于第一阶段训练的训练集输入二分类网络，采用随机梯度下降算法优化softmax损失函数，得到训练后的二分类网络，此训练后的二分类网络中包括训练好的特征提取模块；由于此分类网络只处理二分类问题，而且训练集中仅包含小尺寸图像，所以在第一阶段训练中仅需很小的计算代价便可获得较好的初始化特征提取模块；步骤(3)基于步骤(1)生成的第二阶段训练所需的训练集进行第二阶段训练，得到训练后的PiaNet网络，包含如下子步骤：步骤(3A)将步骤(2A)训练得到的训练好的特征提取模块载入到PiaNet网络中；步骤(3B)将步骤(1C)获得的用于第二阶段训练的训练集输入到PiaNet网络，通过特征提取模块中的放大特征路径获得不同尺度的特征图；步骤(3C)对步骤(3B)输出的不同尺度的特征图进行多尺度锚点设计，并将特征图划分到多尺度锚点中；步骤(3D)步骤(3C)中的每个锚点对标签中相应的ground‑truth检测框进行回归计算，同时对检测框进行优化分类，输出检测框的位置信息和分类结果，采用随机梯度下降算法优化多任务损失函数，得到训练后的PiaNet网络；测试阶段，即步骤(4)精确检测目标，具体为：步骤(4A)将步骤(1C)获得的测试集输入到步骤(3D)训练得到的PiaNet网络中，通过多任务损失函数输出检测框的位置信息和分类结果；步骤(4B)将步骤(4A)中的回归和分类结果输出。