[发明专利]一种基于深度神经网络的吸烟行为检测方法和系统

说明书

本发明公开了一种基于深度神经网络的吸烟行为检测方法和系统。该方法应用于智能巡检机器人，包括以下步骤：智能巡检机器人在自主巡检过程中，采集行人图像，并将所述行人图像输入至基于深度卷积神经网络的吸烟行为检测模型，所述吸烟行为检测模型输出吸烟行为检测结果；所述吸烟行为检测模型为经过轻量化处理和推理加速处理的深度卷积神经网络模型；当所述检测结果为存在吸烟行为时，所述智能巡检机器人发出语音提示。本发明技术方案提升了巡检机器人检测出吸烟行为的速度。

技术领域

本发明涉及吸烟行为检测技术领域，尤其涉及一种基于深度神经网络的吸烟行为检测方法和系统。

背景技术

吸烟行为检测是智能巡检机器人的核心功能，其实现包括以下步骤：首先利用巡检机器人自带的摄像头获取图片，然后基于深度卷积神经网络对获取图片中的吸烟行为进行检测，最后对正在吸烟的行人进行提醒。深度卷积神经网络的吸烟行为检测推理能力直接决定了智能巡检机器人的吸烟行为检测性能优劣。为了保证深度卷积神经网络在推理吸烟图片时具有优良性能，通常先将深度卷积神经网络在公开的吸烟行为数据集上进行预训练，然后再将预训练后的模型放在从实际应用场景采集到的吸烟行为数据集上进行训练。比如，如果生产的智能巡检机器人将会应用于火车站广场区域，那么从实际应用场景采集到的吸烟行为数据集是指在火车站广场上实际采集到的吸烟行为图片组成的数据集。

在机器人工程项目中，模型性能的好坏直接影响巡检机器人的检测效果。所以，在巡检机器人巡检时，模型必须具有良好的鲁棒性和稳定性。基于深度神经网络训练的模型一般具有较好的稳定性，便于在机器人的硬件上部署。视觉注意力机制的引入能够很好的提升模型的鲁棒性，注意力机制通过对神经网络中输入的特征图片进行选择性学习，从而能够快速准确地完成特征学习。比如，在火车站广场采集到的吸烟行为数据集中，图片中大部分的内容可能都不是神经网络所关注的信息，神经网络重点关注的是图片中行人的吸烟行为信息，视觉注意力机制可以让机器人重点关注行人的吸烟行为信息，因此巡检机器人便能够快速准确地检测出行人是否存在吸烟行为。

现有的吸烟检测行为可分为，基于环境传感器的方法，基于可穿戴设备的方法，和基于图像特征检测的方法，其中基于图像特征检测的方法又可分为基于传统的图像特征的方法和基于深度学习的方法。

基于环境传感器的方法通过烟雾传感器检测是否有烟雾产生。当检测到有烟雾时，传感器发生警报。此技术存在的问题如下：(1)吸烟产生的烟雾浓度较低，需要高灵敏的传感器；(2)空气中还有其余杂质，容易产生误报，且不适合广场等拥挤人群。

基于可穿戴设备的方法使用穿戴手表等采集身体运动数据与吸烟行为进行模式匹配，日常活动众多易产生误报，且吸烟者并未同意穿戴检测设备进行吸烟检测，且成本较高。

基于传统的图像特征检测方法利用目标提取方法，通过提取RGB、HSV等颜色空间下的烟雾和香烟特征，和提取目标物体的纹理颜色形状等外观特征，对特征进行判断，检测有无吸烟行为。传统的图像检测方法的难点在于特征的选择，不同环境下，图像的特征不同，如光照强度等。进而降低了传统检测方法的准确度。

发明内容

本发明提供一种基于深度神经网络的吸烟行为检测方法和系统，提升了巡检机器人检测出吸烟行为的速度。

本发明一实施例提供一种基于深度神经网络的吸烟行为检测方法，应用于智能巡检机器人，包括以下步骤：

智能巡检机器人在自主巡检过程中，采集行人图像，并将所述行人图像输入至基于深度卷积神经网络的吸烟行为检测模型，所述吸烟行为检测模型输出吸烟行为检测结果；所述吸烟行为检测模型为经过轻量化处理和推理加速处理的深度卷积神经网络模型；

当所述检测结果为存在吸烟行为时，所述智能巡检机器人发出语音提示。

进一步的，建立基于深度卷积神经网络的初始吸烟行为检测模型，并在所述初始吸烟行为检测模型的每个检测头嵌入CBAM注意力机制，得到第一吸烟行为检测模型，并训练至收敛；