基于预测层特征增强的目标检测方法、存储介质及设备

    公开(公告)号:CN110569887A

    公开(公告)日:2019-12-13

    申请号:CN201910770649.4

    申请日:2019-08-20

    IPC分类号: G06K9/62 G06N3/04 G06N3/08

    摘要: 本发明提供一种基于预测层特征增强的目标检测方法、存储介质及设备,所述方法包括以下步骤:S1、获取需要进行检测的目标图像;S2、通过主干网络提取所述目标图像的特征,得到多个不同尺度的第一预测特征层;S3、将多个不同尺度的所述第一预测特征层进行融合,得到第二预测特征层;S4、构建不同尺寸的锚点框,并将所述第二预测特征层进行第一步锚点框粗略筛选和回归;S5、将所述第二预测特征层进行第二步锚点框精细筛选和回归;S6、输出所述目标图像的检测结果。根据本发明实施例的方法,能够在不增加网络参数的条件下充分发挥网络各特征层的作用的效果,实现对已有网络特征的充分利用。

    基于SE模块的SSD目标检测方法及计算机存储介质

    公开(公告)号:CN110543879A

    公开(公告)日:2019-12-06

    申请号:CN201910769868.0

    申请日:2019-08-20

    IPC分类号: G06K9/32 G06N3/04

    摘要: 本发明提供一种基于SE模块的SSD目标检测方法及计算机存储介质,所述方法包括以下步骤:S1、获取需要进行目标识别的图片或者视频;S2、将卷积神经网络ResNet18的第一个卷积层替换为3×3卷积层,并在ResNet18的第一个和第二个残差块中添加SE模块,形成SE-ResNet18网络结构;S3、将SSD目标检测算法中的主干网络替换为所述SE-ResNet18网络结构,得到检测模型;S4、对所述检测模型进行针对小目标检测的训练,获得训练好的深度神经网络模型;S5、根据训练好的深度神经网络模型对所述图片或者视频的小目标进行检测,得到检测结果。根据本发明实施例的方法,既保证了一定的检测速度,又提升了小目标的检测精度,同时尽量降低了对模型大小的影响。