一种基于深度学习算法的仓储异常目标检测方法及系统.pdf

(19)中华人民共和国国家知识产权局(12)发明专利申请(10)申请公布号CN113850242A(43)申请公布日2021.12.28(21)申请号202111437250.8G06N3/04(2006.01)(22)申请日2021.11.30G06N3/08(2006.01)(71)申请人北京中超伟业信息安全技术股份有限公司地址102200北京市昌平区科技园区超前路甲1号10号楼302室(72)发明人李冠蕊罗远哲刘瑞景荣云虎陆立军郑玉洁李文静王传程刘辉(74)专利代理机构北京高沃律师事务所11569代理人杜阳阳(51)Int.Cl.G06K9/00(2006.01)G06K9/46(2006.01)G06K9/62(2006.01)权利要求书3页说明书10页附图4页(54)发明名称一种基于深度学习算法的仓储异常目标检测方法及系统(57)摘要本发明涉及一种基于深度学习算法的仓储异常目标检测方法及系统，该方法包括：构建基于深度学习算法的仓储异常目标检测网络；以数据增加处理后的仓储视频监控图像为训练集，训练仓储异常目标检测网络，获得仓储异常目标检测模型；利用仓储异常目标检测模型对待检测仓储视频监控图像进行异常目标检测；仓储异常目标检测网络包括依次连接的骨干网络、多尺度特征融合网络和预测层；骨干网络为改进的ResNet34残差网络，改进的ResNet34残差网络的激活函数为Mish激活函数，改进的ResNet34残差网络包括批量标归一化层。本发明提高了仓储异常目标检测的效率和准确性。CN113850242ACN113850242A权利要求书1/3页1.一种基于深度学习算法的仓储异常目标检测方法，其特征在于，包括：获得多个仓储视频监控图像；对多个所述仓储视频监控图像进行数据增加处理，获得数据增加处理后的仓储视频监控图像；构建基于深度学习算法的仓储异常目标检测网络；以数据增加处理后的仓储视频监控图像为训练集，训练所述仓储异常目标检测网络，获得仓储异常目标检测模型；利用所述仓储异常目标检测模型对待检测仓储视频监控图像进行异常目标检测；所述仓储异常目标检测网络包括依次连接的骨干网络、多尺度特征融合网络和预测层；所述骨干网络为改进的ResNet34残差网络，所述改进的ResNet34残差网络的激活函数为Mish激活函数，所述改进的ResNet34残差网络包括批量标归一化层；所述多尺度特征融合网络包括FPN网络和PAN网络，所述FPN网络用于对所述改进的ResNet34残差网络输出的各特征图进行上采样，获得第一特征图像金字塔模型，所述PAN网络用于对所述FPN网络输出的各特征图进行下采样，获得第二特征图像金字塔模型；所述预测层采用的损失函数为综合损失函数，所述综合损失函数为回归框损失函数、交叉熵损失函数和IOU置信度损失函数之和。2.根据权利要求1所述的基于深度学习算法的仓储异常目标检测方法，其特征在于，所述深度学习算法为YOLOv5算法。3.根据权利要求1所述的基于深度学习算法的仓储异常目标检测方法，其特征在于，所述对多个所述仓储视频监控图像进行数据增加处理，获得数据增加处理后的仓储视频监控图像，具体包括：采用对所述仓储视频监控图像的亮度、对比度、色调、饱和度和高斯噪声进行改变，或者对所述仓储视频监控图像进行裁剪、翻转、旋转或随机缩放，获得数据增加处理后的仓储视频监控图像。4.根据权利要求1所述的基于深度学习算法的仓储异常目标检测方法，其特征在于，所述改进的ResNet34残差网络包括多个依次连接的特征提取模块，各所述特征提取模块包括第一特征提取单元和第二特征提取单元，所述第一特征提取单元的输入为所述特征提取模块的输入，所述第一特征提取单元的输出连接所述第二特征提取单元的输入，所述第二特征提取单元的输出为所述特征提取模块的输出，所述第一特征提取单元和第二特征提取单元均包括依次连接的卷积层、批量标归一化层和Mish激活函数，所述特征提取模块的输入连接所述特征提取模块的输出。5.根据权利要求1所述的基于深度学习算法的仓储异常目标检测方法，其特征在于，所述预测层采用K‑Means聚类算法确定锚框，所述K‑Means聚类算法中k值为6，采用非极大值抑制算法进行锚框选择。6.根据权利要求1所述的基于深度学习算法的仓储异常目标检测方法，其特征在于，所述综合损失函数表示为：Loss=LossB+LossC+LossO；其中，LossB表示回归框损失函数，LossC表示交叉熵损失函数，LossO表示IOU置信度损失函数；2CN113850242A权利要求书2/3页；；；其中，λCoord表示第一超参数，λCLS表示第二超参数，λNoo表示第三超参数，λo表示第四超2参数，B表示偏置参数，S表示网格中单元格的数量，xi表示单元格i中心位置的x轴坐标，yi表示