[发明专利]检测客车内的安全设施的方法、设备和存储介质

权利要求书

1.一种用于检测客车内的安全设施的方法，包括：

获取客车内的多张原始图片；

检测所述多张原始图片以从所述多张原始图片中获取包括所述安全设施的多个定位子图；

对获取的多个定位子图进行分类，以从所述定位子图中分类出属于目标安全设施的第一集合；

利用分割模型对所述第一集合中的每张定位子图进行分割以得到每张定位子图的灰度图并确定所述灰度图中构成所述目标安全设施的像素，并且将所述定位子图中构成所述目标安全设施的像素置为预定值以得到背景图片的第二集合；以及

利用谱聚类的方法将所述第二集合的背景图片聚类为多个簇，其中所述簇的数量指示所述客车内的目标安全设施的数量。

2.如权利要求1所述的方法，其中检测所述多张原始图片包括：：

在SSD检测模型中设定分数阈值和个数阈值；

将所述多张原始图片输入所述SSD检测模型，以计算满足所述分数阈值的定位子图的个数；

如果满足所述分数阈值的定位子图的个数小于所述个数阈值，则返回所有满足所述分数阈值的定位子图作为所述多个定位子图；以及

如果满足所述分数阈值的定位子图的个数大于或等于所述个数阈值，则从所有满足所述分数阈值的定位子图中选择所述个数阈值个定位子图作为所述多个定位子图。

3.如权利要求1所述的方法，其中对获取的多个定位子图进行分类，以从所述定位子图中分类出属于所述目标安全设施的定位子图的第一集合包括：

将所述多个定位子图中的每个定位子图输入ResNet分类模型，以得到所述定位子图属于所述目标安全设施的类别的得分和不属于所述目标安全设施的类别的得分，以及

如果属于所述目标安全设施的类别的得分高于不属于所述目标安全设施的类别的得分，则将所述定位子图放入所述第一集合。

4.如权利要求1所述的方法，其中将所述定位子图中构成所述目标安全设施的像素置为预定值以得到背景图片的第二集合包括：

确定所述灰度图中的每个像素的灰度值是否为预定灰度值，其中所述预定灰度值通过对所述分割模型进行训练所确定；

如果确定所述像素的灰度值是预定灰度值，则确定所述像素是构成所述目标安全设施的像素；

将所述定位子图中构成所述目标安全设施的所有像素设置为所述预定值以得到更新的定位子图；

将所述更新的定位子图作为背景图片放入所述第二集合。

5.如权利要求4所述的方法，其中所述预定值包括0或255。

6.如权利要求1所述的方法，其中利用谱聚类的方法将所述第二集合的背景图片聚类为多个簇包括：

将所述第二集合的每张背景图片分别输入到孪生神经网络中以得到每张背景图片的特征向量x_i；以及

对第二集合的所有背景图片的特征向量x_i进行矩阵变换，并基于谱聚类的方法将变换后产生的样本点聚类为多个簇。