[发明专利]行人吸烟识别模型构建、识别方法、装置及电子设备

权利要求书

1.一种行人吸烟识别模型构建方法，其特征在于，所述方法包括：

收集行人图像，对所述行人图像中行人对应的部分图像进行标记，获取第一标记信息，将所述行人图像和所述第一标记信息，构成第一数据集；

收集吸烟行为图像，对用以指示所述吸烟行为的动作图像进行标记，获取第二标记信息，将所述吸烟行为图像和所述第二标记信息构成第二数据集；

收集携带有背景信息的香烟图像，对所述香烟图像进行图像处理后，与所述背景信息生成新的图像；

对所述新的图像中香烟图像进行标记，获取第三标记信息，将所述新的图像和所述第三标记信息，构成第三数据集；

利用所述第一数据集对第一目标检测模型进行训练，得到行人检测模型；

利用所述第二数据集所述第三数据集，对第二目标检测模型进行训练，得到吸烟检测模型；

根据所述行人检测模型和所述吸烟检测模型，构建行人吸烟识别模型。

2.根据权利要求1所述的方法，其特征在于，所述利用所述第二数据集所述第三数据集，对第二目标检测模型进行训练，得到吸烟检测模型，具体包括：

合并所述第二数据集和所述第三数据集，构成第四数据集；

利用第四数据集对所述第二目标检测模型进行训练，获取吸烟检测模型。

3.一种行人吸烟识别方法，其特征在于，所述方法由如权利要求1或2任一项所述的行人吸烟识别模型执行，所述方法包括：

获取监控图像，利用行人检测模型对所述监控图像进行识别；

当确定所述监控图像中包括行人时，利用吸烟检测模型对所述行人的行为图像进行识别，以确定所述行人是否存在吸烟行为。

4.根据权利要求3所述的方法，其特征在于，当确定所述监控图像中包括行人时，利用吸烟检测模型对所述行人的行为图像进行识别，具体包括：

识别所述行为图像中的是否存在用以指示吸烟行为的动作图像；

当确定所述行为图像中包括用以指示吸烟行为的动作图像后，确定包括所述动作图像的行为图像的数量；

根据所述行为图像中包括所述动作图像的数量，确定所述行人是否存在吸烟行为。

5.根据权利要求4所述的方法，其特征在于，所述动作图像包括：香烟，行人的头和/或手与所述香烟之间具有预设位置关系；

当确定所述动作图像中包括香烟，且确定所述行人的头和/或手，与所述香烟之间具有预设位置关系时，确定行为图像中存在用以指示吸烟行为的图像。

6.根据权利要求5所述的方法，其特征在于，所述行人的头和/或手，与所述香烟之间具有预设位置关系，具体包括：

所述行人的头的位置和/或所述行人的手的位置，与所述香烟的位置相交。

7.一种行人吸烟识别模型构建装置，其特征在于，所述装置包括：

收集模块，用于收集行人图像；

标记模块，用于对所述行人图像中行人对应的部分图像进行标记，获取第一标记信息；

处理模块，用于将所述行人图像和所述第一标记信息，构成第一数据集；

所述收集模块，还用于收集吸烟行为图像；

所述标记模块，还用于对用以指示所述吸烟行为的动作图像进行标记，获取第二标记信息；

所述处理模块，还用于将所述吸烟行为图像和所述第二标记信息构成第二数据集；

所述收集模块，还用于收集携带有背景信息的香烟图像；

所述处理模块，还用于对所述香烟图像进行图像处理后，与所述背景信息生成新的图像；

所述标记模块，还用于对所述新的图像中香烟图像进行标记，获取第三标记信息；

所述处理模块，还用于将所述新的图像和所述第三标记信息，构成第三数据集；利用所述第一数据集对第一目标检测模型进行训练，得到行人检测模型；利用所述第二数据集所述第三数据集，对第二目标检测模型进行训练，得到吸烟检测模型；根据所述行人检测模型和所述吸烟检测模型，构建行人吸烟识别模型。