[发明专利]一种安全检测的方法和装置

权利要求书

1.一种安全检测的方法，其特征在于，包括：

利用训练好的目标检测模型，检测图片中的检测对象；

当所述目标检测模型检测的结果指示存在至少一种所述检测对象时，利用训练好的分类模型检测至少一种所述检测对象佩戴安全防护设备的情况；

根据检测的结果，为至少一种所述检测对象生成安全防护信息。

2.根据权利要求1所述的安全检测的方法，其特征在于，进一步包括：

获取第一样本集合，所述第一样本集合中存在标注有第一种参考对象和参考目标的多个第一样本图片，且所述第一样本集合中存在标注有第二种参考对象的多个第二样本图片，其中，所述第一种参考对象与所述参考目标之间存在关联关系；

利用标注好的所述第一样本集合，训练目标检测模型，得到训练好的所述目标检测模型。

3.根据权利要求2所述的安全检测的方法，其特征在于，

利用训练好的所述目标检测模型以及设定检测阈值，从所述第一样本集合选取不小于所述检测阈值的多个所述第一样本图片和/或多个所述第二样本图片；

将选取的多个所述第一样本图片和/或多个所述第二样本图片组成第二样本集合；

基于所述第二样本集合中所述第一种参考对象和/或所述第二种参考对象佩戴安全防护设备的情况以及所述第二样本集合，构建一个或多个分类样本集合；

利用所述分类样本集合训练分类模型，得到训练好的分类模型。

4.根据权利要求1所述的方法，其特征在于，检测图片中的检测对象，包括：

识别检测图片中的检测目标；

当识别的结果指示所述图片包括第一种检测目标；

判断是否存在与所述第一种检测目标存在关联关系的第二种检测目标，如果是，则确定所述第一种检测目标为第一种检测对象；否则，确定所述第一种检测目标为第二种检测对象。

5.根据权利要求2所述的方法，其特征在于，

所述目标检测模型设置有检测分支；

训练所述目标检测模型，包括：

利用所述第一种参考对象与所述参考目标之间存在关联关系，训练所述检测分支，以利用训练好的检测分支检测出第一种检测对象，所述第一种检测对象与所述第一种参考对象相匹配。

6.根据权利要求5所述的方法，其特征在于，

所述关联关系包括：所述第一种参考对象包括的多个第一关键点的坐标和所述第一关键点的状态，以及所述参考目标包括的第二关键点的坐标和所述第二关键点的状态，其中，第一关键点与所述参考目标存在关联，所述第二关键点与所述第一种参考对象存在关联；

利用多个第一关键点的坐标和所述第一关键点的状态以及所述第二关键点的坐标和所述第二关键点的状态，训练所述检测分支。

7.根据权利要求3所述的方法，其特征在于，进一步包括：

获取第三样本集合，所述第三样本集合包含第一种参考对象、第二种参考对象、参考目标中的任意一个或多个，其中，每一种所述参考对象佩戴有一个或多个安全防护设备；

基于所述第二样本集合和所述第三样本集合，执行构建一个或多个分类样本集合的步骤。

8.根据权利要求7所述的方法，其特征在于，构建一个或多个分类样本集合，包括：

从所述第二样本集合和所述第三样本集合中选取设定数量的目标样本图片，对各个目标样本图片包含的一个或多个所述安全防护设备添加标注信息；

利用所述各个目标样本图片和对应的所述标注信息，将所述设定数量的目标样本图片划分至一个或多个分类样本集合中。

9.根据权利要求8所述的方法，其特征在于，训练分类模型，包括：

利用所述分类样本集合、以及所述第二样本集合和所述第三样本集合中未标注的其他训练样本图片，训练所述分类模型。