一种基于注意力图的吸烟行为检测方法及装置

摘要

本申请提供一种基于注意力图的吸烟行为检测方法及装置，该方法包括：提取目标图像的人脸区域，得到人脸图像；利用特征提取网络获取该人脸图像的第一特征图；对第一特征图进行卷积运算，得到M个注意力图，每个注意力图对应于人脸图像上的一个注意力集中区域；将M个注意力图作用到第一特征图中，得到第二特征图；根据第二特征图确定人脸图像中人物是否存在吸烟行为。上述方法首先检测人脸区域，再基于人脸区域的图像特征进行分类，由于烟在人脸区域所占的比例较大，所以，可以有效提取到烟的特征，避免了由于烟过小而检测不到的问题，同时，基于注意力机制对烟出现概率大的位置进行特征增强，有利于提升吸烟行为检测的准确度。