利索能及
我要发布
收藏
专利号: 2019111888955
申请人: 江南大学
专利类型:发明专利
专利状态:已下证
更新日期:2025-10-27
缴费截止日期: 暂无
联系人

摘要:

权利要求书:

1.一种基于密集连接深度网络的目标检测方法,其特征在于,包括以下步骤:步骤(1):读入Pascal VOC数据集中的图像数据并提取目标数据特征;

步骤(2):训练网络模型;

步骤(3):进行目标检测;

具体的步骤为:

步骤(1):读入Pascal VOC数据集中的图像数据并提取目标数据特征:网络读取输入的图像数据,首先将其分辨率归一化为416*416,然后通过一个卷积层和一个池化层后得到输出大小为208*208的特征映射图,再经过5个密集连接模块对待检测图像进行特征提取,得到13x13大小的特征图,并将提取得到的13x13大小的特征图进行上采样,得到26x26大小的特征映射图;其中卷积层中卷积核大小为3*3,步长为1;池化层中池化的核大小为2*2,步长为2;

步骤(2):训练网络模型:设置网络batch为64,重复迭代训练得到检测模型;

步骤(3):进行目标检测:网络首先通过特征提取网络对输入图像提取特征,得到一定尺寸的k*k特征图,然后将输入图像分为k*k个单元格,每个单元格预测固定数量的边界框;

预测时采用逻辑回归,用于预测每个边界框的目标性得分,即这块区域是目标的可能性有多大;而后进行非极大值抑制NMS,最后输出检测结果;

所述步骤(1)还包括:

①引入的密集连接方式,使得L层网络有L(L+1)/2个连接;其中的密集连接模块Dense Block主要是由1*1和3*3的卷积层组成,其中的1*1卷积操作又被成为bottleneck layer;

3*3卷积则是用于提取图像特征;密集连接模块Dense Block中每一层的输入来自前面所有层的输出;下式表明了第l层的输入即为之前所有层的输出之和;

xl=Hl([x0,x1,…,xl‑1])其中,xl表示第l层的输出,[x0,x1,L,xl‑1]表示第0,L,l‑1层输出的级联;上式中Hl(g)表示三个连续运算的复合函数,由BN、ReLU和一个3*3的卷积层组成;

②减少密集连接模块Dense Block中卷积层输出特征图数量;其中,Dense Block1设定特征映射图数量为16,Dense Block2、Dense Block3、Dense Block4、Dense Block5设定为

32、64、128、256;输出特征图数量不断递增的目的是使网络能够学习到图像数据中更为丰富的高层语义信息,增加定位的精确度;

③将网络分为多个密集连接模块Dense Block,其中,设定不同Dense Block的特征图数量均不相同且每个Dense Block的特征图输出数量都呈倍数增长,数量分别为16,32,64,

128,256,并将每个Dense Block中卷积得到特征图的输出大小都设定为相同;

所述步骤(2)包括:

设置网络的学习率为0.001,动量设为0.9,权重衰减正则项为0.0005,网络最大的迭代次数为500200,网络的学习率在迭代次数到达400000和450000时衰减10倍;同时网络使用多尺度训练,网络读取数据后,图像归一化分辨率的宽高在320~608之间取随机值,且每10轮随机改变一次,随机值均为32的倍数;

所述步骤(3)包括:

①yolov3‑tiny使用K‑means聚类算法聚类数据集中真实框,为步骤(1)中得到的两种尺度13x13、26x26的特征图设定3个不同大小的先验框,总共聚类出6个不同尺寸的先验框;

两种不同尺度的6个先验框大小如下所示:②使用6个不同先验框Anchors在两种不同尺度13x13、26x26的特征映射图上进行预测;在对边界框进行预测的时候,为了更好的数据建模以及支持多标签分类,网络采用逻辑回归logistic regression;网络边界框的坐标预测公式如下:bx=σ(tx)+cx

by=σ(ty)+cy

其中tx、ty、tw、th为模型的实际预测值,cx和cy表示grid cell的坐标偏移量,pw和ph表示anchor box的宽高,bx、by、bw和bh为最终得到的边界框的中心的坐标和宽高;坐标的训练采用的是平方和误差损失;

③设定非极大值抑制NMS的阈值为0.45。

2.权利要求1所述的基于密集连接深度网络的目标检测方法在图像目标检测中的应用。

3.根据权利要求2所述的应用,其特征在于,具体应用步骤为:网络读取不同场景下的行人图像数据作为训练数据,首先将图像数据的图像分辨率归一化为416*416,而后经过一个卷积层和一个池化层得到输出大小为208*208的特征映射图,再经过5个密集连接模块对待检测图像进行特征提取,得到13x13大小的特征图,并将提取得到的13x13大小的特征图进行上采样,得到26x26大小的特征映射图;其中卷积层中卷积核大小为3*3,步长为1;池化层中池化的核大小为2*2,步长为2;通过训练网络得到相应的检测模型;加载训练得到的模型、网络配置文件以及待检测的图像,网络首先通过特征提取网络对输入的待检测图像提取特征;提取特征后得到13*13、26*26的特征图,并在这两种不同尺度下进行预测;随后网络将待检测图像分别划分为13*13、26*26个单元格,每个单元格预测固定3个的边界框;预测时采用逻辑回归,用于预测每个边界框的目标性得分,即这块区域是行人类别的可能性有多大;而后进行非极大值抑制NMS,最后输出检测结果。

4.权利要求1所述的基于密集连接深度网络的目标检测方法在视频目标检测中的应用。

5.根据权利要求4所述的应用,其特征在于,具体应用步骤为:网络读取不同场景下的视频目标图像数据作为训练数据,首先将目标图像数据的图像分辨率归一化为416*416,再经过5个密集连接模块对待检测图像进行特征提取,得到13x13大小的特征图,并将提取得到的13x13大小的特征图进行上采样,得到26x26大小的特征映射图;其中卷积层中卷积核大小为3*3,步长为1;池化层中池化的核大小为2*2,步长为2;通过训练网络得到相应检测任务的检测模型;加载训练得到的模型、网络配置文件以及待检测的图像,网络首先通过特征提取网络对输入的待检测图像提取特征;提取特征后得到13*13、26*26的特征图,并在这两种不同尺度下进行预测;随后网络将待检测图像分别划分为13*13、26*26个单元格,每个单元格预测固定3个的边界框;预测时采用逻辑回归,用于预测每个边界框的目标性得分,即这块区域是目标类别的可能性有多大;而后进行非极大值抑制NMS;最后输出检测结果。