一种带有区域放大操作的改进SSD目标检测算法的制作方法

文档序号:20954277发布日期:2020-06-02 20:19阅读:来源:国知局

技术特征:

1.一种带有区域放大操作的改进ssd目标检测算法,步骤如下:

(1)采用ssd网络,在公开数据集pascalvoc2007和pascalvoc2012上训练获得ssd模型;

(2)在ssd网络的低层特征图采用区域放大策略,利用步骤(1)中训练获得的ssd模型生成目标对应的建议区域,并通过区域映射操作将建议区域映射到相应低层特征层上,对建议区域对应的目标特征进行放大以提高小目标的检测精度;

(3)通过特征提取操作,对ssd网络的多个高层进行特征提取,获得新的网络层,增加网络深度改善中目标的检测精度;

(4)将改进的ssd网络低层特征和高层特征进行结合,重构特征金字塔,并利用类别和位置检测层实现目标检测;

(5)根据改进方法,在步骤(1)中训练获得的ssd模型基础上进行参数再训练以获得最终的模型。

2.根据权利要求1所述的一种带有区域放大操作的改进ssd目标检测算法,其特征在于,步骤(1)中是从公开数据集pascalvoc2007和pascalvoc2012中挑选7类目标组成训练数据集,包括16237张图像,检测类别为自行车、公交车、汽车、猫、狗、摩托车和人。

3.根据权利要求1所述的一种带有区域放大操作的改进ssd目标检测算法,其特征在于,步骤(2)中利用ssd网络用于检测的低层卷积层conv4_3,对其采用反卷积操作获得高分辨反卷积图,并将建议区域映射到反卷积图中,对目标区域特征进行放大,丰富特征细节信息。另外,额外引入更低层卷积层conv3_3,通过特征层提取操作获得对应特征图,同样利用建议区域映射实现目标区域特征放大。

4.根据权利要求1所述的一种带有区域放大操作的改进ssd目标检测算法,其特征在于,该算法训练步骤如下:

(1)采用ssd网络,在公开数据集pascalvoc数据集上训练获得ssd模型,迭代12万次,四块gpu,型号为nvidiatitanxp,显存12gb。

(2)从pascalvoc数据集中重新挑选训练集,包括3376张图像,由于mscoco数据集中的中目标和小目标相对较多,在mscoco数据集中挑选不同尺度的小、中目标组成测试集。

(3)使用步骤(2)的训练集,在步骤(1)中训练的ssd模型参数的基础上,采用改进方法进行训练获得最终模型。

5.根据权利要求4所述的一种带有区域放大操作的改进ssd目标检测算法,其特征在于,步骤(3)中由于在已训练好ssd模型已有参数基础上进行再训练,改进方法训练集有所减少,且在一块型号为gtx1060、显存为6gb的gpu上即可完成训练与测试,降低了训练对gpu的成本和计算能力的要求。

6.根据权利要求4所述的一种带有区域放大操作的改进ssd目标检测算法,其特征在于,步骤(3)中网络超参数设置为:初始学习速率为0.0001,权重衰减为0.0005,动量为0.9,采用随机梯度下降算法训练网络,迭代47600次收敛获得最终模型。


技术总结
本发明涉及一种带有区域放大操作的改进SSD目标检测算法,步骤如下:采用SSD网络,在公开数据集上训练获得SSD模型;在SSD网络低层引入区域放大操作,利用SSD网络生成建议区域,并将建议区域映射到低层特征层上,对建议区域特征进行放大以提高小目标的检测精度;对SSD网络的高层进行特征提取构成新的网络层,增加网络深度以改善中目标的检测精度;将改进的低层和高层进行结合重构特征金字塔,利用类别和位置检测层进行检测;在SSD模型基础上进行再训练以获得最终模型。本发明引入区域放大操作,在SSD网络的低层和高层采用不同改进策略,以平均精度均值mAP作为评价标准,提高对小、中目标的检测精度。

技术研发人员:王以忠;陈幻杰;郭肖勇;王琦琦;杨国威
受保护的技术使用者:天津科技大学
技术研发日:2019.05.20
技术公布日:2020.06.02
当前第2页1 2 
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1