吸烟图像的识别方法及装置、存储介质、电子设备制造方法及图纸

本发明专利技术公开了一种吸烟图像的识别方法及装置、存储介质、电子设备，属于人工智能领域。其中，该方法包括：获取待识别的目标图像；采用第一预测模型预测所述目标图像，得到第一识别结果和所述第一识别结果的第一置信度，采用第二预测模型预测所述目标图像，得到第二识别结果和所述第二识别结果的第二置信度；基于所述第一置信度和所述第二置信度输出所述目标图像的第三识别结果，其中，所述第三识别结果用于表征所述目标图像内是否包括吸烟场景。通过本发明专利技术，全局识别和局部识别相结合，较好地保持模型召回率的同时，提高识别结果的精确率，解决了相关技术中识别吸烟图像的准确率低的技术问题。

【技术实现步骤摘要】
吸烟图像的识别方法及装置、存储介质、电子设备
本专利技术涉及人工智能领域，具体而言，涉及一种吸烟图像的识别方法及装置、存储介质、电子设备。
技术介绍
相关技术中，需要在室内或某些公共场合通过图像检测吸烟场景，判断场所是否有人在吸烟，进而报警。吸烟图像识别通常采用深度学习的方法对图像进行分类或检测图像中烟的位置进行识别；整个过程需要采集数据、标注数据、训练模型、模型推理(即用训练好的模型进行预测)。相关技术中，采用分类算法进行检测识别，需要标注大量数据才能训练出较好的模型，使用图像分类的方法能召回较多吸烟图像，但容易误判，对训练集中相似吸烟背景的图像产生误判，导致精确率较低。针对相关技术中存在的上述问题，目前尚未发现有效的解决方案。
技术实现思路
本专利技术实施例提供了一种吸烟图像的识别方法及装置、存储介质、电子设备。根据本专利技术实施例的一个方面，提供了一种吸烟图像的识别方法，包括：获取待识别的目标图像；采用第一预测模型预测所述目标图像，得到第一识别结果和所述第一识别结果的第一置信度，采用第二预测模型预测所述目标图像，得到第二识别结果和所述第二识别结果的第二置信度，其中，所述第一预测模型用于从全局识别所述目标图像内是否包括吸烟场景，所述第二预测模型用于从局部识别所述目标图像内是否包括吸烟场景；基于所述第一置信度和所述第二置信度输出所述目标图像的第三识别结果，其中，所述第三识别结果用于表征所述目标图像内是否包括吸烟场景。进一步，基于所述第一置信度和所述第二置信度输出所述目标图像的第三识别结果包括：判断所述第一置信度和所述第二置信度是否符合预设条件；若所述第一置信度和所述第二置信度符合预设条件，判断所述第一识别结果和所述第二识别结果是否均为正例结果，其中，所述正例结果用于表征所述目标图像内包括吸烟场景；若所述第一识别结果和所述第二识别结果均为正例结果，输出所述目标图像的第三识别结果为所述正例结果。进一步，判断所述第一置信度和所述第二置信度是否符合预设条件包括：判断所述第一置信度是否大于或等于第一阈值，或，判断所述第二置信度是否大于或等于第二阈值；若所述第一置信度大于或等于第一阈值，或所述第二置信度大于或等于第二阈值，确定所述第一置信度和所述第二置信度符合预设条件。进一步，判断所述第一置信度和所述第二置信度是否符合预设条件包括：判断所述第一置信度是否大于或等于第三阈值且小于第四阈值，判断所述第二置信度是否大于或等于第五阈值且小于第六阈值；若所述第一置信度大于或等于第三阈值且小于第四阈值，且，所述第二置信度大于或等于第五阈值且小于第六阈值，确定所述第一置信度和所述第二置信度符合预设条件。进一步，在采用第一预测模型预测所述目标图像之前，所述方法还包括：获取第一样本数据，其中，所述第一样本数据包括正样本和负样本；采用所述第一样本数据训练二分类模型，得到所述第一预测模型。进一步，在采用第二预测模型预测所述目标图像之前，所述方法还包括：获取第一样本数据，其中，所述第一样本数据包括正样本和负样本，所述正样本包含香烟所在的图像框；采用所述第一样本数据训练基于yolo算法的目标检测模型，得到所述第二预测模型。进一步，在基于所述第一置信度和所述第二置信度输出所述目标图像的第三识别结果之后，所述方法还包括：采集待选样本图像；基于所述第一预测模型和所述第二预测模型从所述待选样本图像中筛选若干个潜在样本图像；设置所述若干个潜在样本图像的标签信息，并对所述标签信息进行校正后，得到第二样本数据；采用所述第二样本数据迭代训练所述第一预测模型和所述第二预测模型。根据本专利技术实施例的另一个方面，提供了一种吸烟图像的识别装置，包括：第一获取模块，用于获取待识别的目标图像；预测模块，用于采用第一预测模型预测所述目标图像，得到第一识别结果和所述第一识别结果的第一置信度，采用第二预测模型预测所述目标图像，得到第二识别结果和所述第二识别结果的第二置信度，其中，所述第一预测模型用于从全局识别所述目标图像内是否包括吸烟场景，所述第二预测模型用于从局部识别所述目标图像内是否包括吸烟场景；输出模块，用于基于所述第一置信度和所述第二置信度输出所述目标图像的第三识别结果，其中，所述第三识别结果用于表征所述目标图像内是否包括吸烟场景。进一步，所述输出模块包括：第一判断单元，用于判断所述第一置信度和所述第二置信度是否符合预设条件；第二判断单元，用于若所述第一置信度和所述第二置信度符合预设条件，判断所述第一识别结果和所述第二识别结果是否均为正例结果，其中，所述正例结果用于表征所述目标图像内包括吸烟场景；输出单元，用于若所述第一识别结果和所述第二识别结果均为正例结果，输出所述目标图像的第三识别结果为所述正例结果。进一步，所述第一判断单元包括：第一判断子单元，用于判断所述第一置信度是否大于或等于第一阈值，或，判断所述第二置信度是否大于或等于第二阈值；第一确定子单元，用于若所述第一置信度大于或等于第一阈值，或所述第二置信度大于或等于第二阈值，确定所述第一置信度和所述第二置信度符合预设条件。进一步，所述第一判断单元包括：第二判断子单元，用于判断所述第一置信度是否大于或等于第三阈值且小于第四阈值，判断所述第二置信度是否大于或等于第五阈值且小于第六阈值；第二确定子单元，用于若所述第一置信度大于或等于第三阈值且小于第四阈值，且，所述第二置信度大于或等于第五阈值且小于第六阈值，确定所述第一置信度和所述第二置信度符合预设条件。进一步，所述装置还包括：第二获取模块，用于在所述预测模块采用第一预测模型预测所述目标图像之前，获取第一样本数据，其中，所述第一样本数据包括正样本和负样本；第一训练模块，用于采用所述第一样本数据训练二分类模型，得到所述第一预测模型。进一步，所述装置还包括：第三获取模块，用于在所述预测模块采用第一预测模型预测所述目标图像之前，获取第一样本数据，其中，所述第一样本数据包括正样本和负样本，所述正样本包含香烟所在的图像框；第二训练模块，用于采用所述第一样本数据训练基于yolo算法的目标检测模型，得到所述第二预测模型。进一步，所述装置还包括：采集模块，用于在所述输出模块基于所述第一置信度和所述第二置信度输出所述目标图像的第三识别结果之后，采集待选样本图像；筛选模块，用于基于所述第一预测模型和所述第二预测模型从所述待选样本图像中筛选若干个潜在样本图像；设置模块，用于设置所述若干个潜在样本图像的标签信息，并对所述标签信息进行校正后，得到第二样本数据；第三训练模块，用于采用所述第二样本数据迭代训练所述第一预测模型和所述第二预测模型。根据本专利技术实施例的另一方面，还提供了一种存储介质，该存储介质包括存储的程序，程序运行时执行上述的步骤。根据本专利技术实施例的另一方面，还提供了一种电子设备，包括处理器、通信接口、存储器和通信总线，其中，处理器，通信接口，存储器通过通信总线完成相互间的通信；其中：存储器，用于存放计算机程序；处理器，用于通过运行存储器上所存放的程序来执行上述方法中的步骤本文档来自技高网...

【技术保护点】
1.一种吸烟图像的识别方法，其特征在于，包括：/n获取待识别的目标图像；/n采用第一预测模型预测所述目标图像，得到第一识别结果和所述第一识别结果的第一置信度，采用第二预测模型预测所述目标图像，得到第二识别结果和所述第二识别结果的第二置信度，其中，所述第一预测模型用于从全局识别所述目标图像内是否包括吸烟场景，所述第二预测模型用于从局部识别所述目标图像内是否包括吸烟场景；/n基于所述第一置信度和所述第二置信度输出所述目标图像的第三识别结果，其中，所述第三识别结果用于表征所述目标图像内是否包括吸烟场景。/n

【技术特征摘要】
1.一种吸烟图像的识别方法，其特征在于，包括：
获取待识别的目标图像；
采用第一预测模型预测所述目标图像，得到第一识别结果和所述第一识别结果的第一置信度，采用第二预测模型预测所述目标图像，得到第二识别结果和所述第二识别结果的第二置信度，其中，所述第一预测模型用于从全局识别所述目标图像内是否包括吸烟场景，所述第二预测模型用于从局部识别所述目标图像内是否包括吸烟场景；
基于所述第一置信度和所述第二置信度输出所述目标图像的第三识别结果，其中，所述第三识别结果用于表征所述目标图像内是否包括吸烟场景。


2.根据权利要求1所述的方法，其特征在于，基于所述第一置信度和所述第二置信度输出所述目标图像的第三识别结果包括：
判断所述第一置信度和所述第二置信度是否符合预设条件；
若所述第一置信度和所述第二置信度符合预设条件，判断所述第一识别结果和所述第二识别结果是否均为正例结果，其中，所述正例结果用于表征所述目标图像内包括吸烟场景；
若所述第一识别结果和所述第二识别结果均为正例结果，输出所述目标图像的第三识别结果为所述正例结果。


3.根据权利要求2所述的方法，其特征在于，判断所述第一置信度和所述第二置信度是否符合预设条件包括：
判断所述第一置信度是否大于或等于第一阈值，或，判断所述第二置信度是否大于或等于第二阈值；
若所述第一置信度大于或等于第一阈值，或所述第二置信度大于或等于第二阈值，确定所述第一置信度和所述第二置信度符合预设条件。


4.根据权利要求2所述的方法，其特征在于，判断所述第一置信度和所述第二置信度是否符合预设条件包括：
判断所述第一置信度是否大于或等于第三阈值且小于第四阈值，判断所述第二置信度是否大于或等于第五阈值且小于第六阈值；
若所述第一置信度大于或等于第三阈值且小于第四阈值，且，所述第二置信度大于或等于第五阈值且小于第六阈值，确定所述第一置信度和所述第二置信度符合预设条件。


5.根据权利要求1所述的方法，其特征在于，在采用第一预测模型预测所述目标图像之前，所述方法还包括：
获取第一样本...

【专利技术属性】
技术研发人员：肖鑫，许健，田波，路洪运，
申请(专利权)人：北京金山云网络技术有限公司，
类型：发明
国别省市：北京;11