一种用于准确检测危险物品的改进型SSD算法

摘要

本发明公开了一种用于准确检测危险物品的改进型SSD算法，包括以下步骤：构建一个基于SSD算法的改进型卷积神经网络，并且初始化网络模型的参数；设定各个超参数所需图片的批次和学习速率，并且设定所需训练的事件次数；通过所构建的数据集对卷积神经网络进行训练,得到权重文件，将训练好的权重文件放到测试模型中，向测试模型中输入实时场景下的图片，将训练好的目标特征与图片中的特征信息进行比对，通过分类器完成目标的分类识别，从而实现对危险物品的检测；本申请通过对现有的SSD算法进行改进，通过改变先验框的尺寸大小和改变conv4_3特征层的处理，从而实现对远距离情况下危险物品的准确检测，满足使用需求。