[发明专利]基于无监督异常检测的溺水检测方法及装置

权利要求书

1.一种基于无监督异常检测的溺水检测方法，其特征在于，包括以下步骤：

获取水下视频流，并对所述水下视频流进行图像增强；

将图像增强后的水下视频流输入到预先训练好的轻量级目标检测模型，以通过所述轻量级目标检测模型判断所述水下视频流中游泳者的人体姿态是否为垂直状态；

如果判断结果为是，则采用深度高斯异常分类法计算该水下视频流对应的游泳者特征向量与正常高斯模型的马氏距离，并将该马氏距离作为异常得分；

判断所述异常得分是否大于预设得分阈值；

如果是，则认为该水下视频流对应的游泳者发生溺水，并生成报警信息。

2.如权利要求1所述的基于无监督异常检测的溺水检测方法，其特征在于，对所述水下视频流进行图像增强，包括：

通过限制对比度自适应直方图均衡化算法对所述水下视频流的RGB三通道分别进行增强，以输出图像增强后的水下视频流。

3.如权利要求1所述的基于无监督异常检测的溺水检测方法，其特征在于，采用深度高斯异常分类法计算该水下视频流对应的游泳者特征向量与正常高斯模型的马氏距离之前，还包括：

获取第一正常游泳者图像数据集，并根据所述第一正常游泳者图像数据集进行训练，以得到第一卷积自编码器，只保留第一卷积自编码器的编码器部分的网络，得到第一编码器网络；

获取第二正常游泳者图像数据集，并将所述第二正常游泳者图像数据集输入到所述第一编码器网络，以通过所述第一编码器网络提取所述第二正常游泳者图像数据集中人体正常样本的特征向量；

根据所述人体正常样本的特征向量构建多元高斯模型，并将所述多元高斯模型作为正常高斯模型；

对所述第一编码器网络进行再次训练，以得到第二编码器网络，以便在将水下视频流输入到所述第二编码器网络中时，通过所述第二编码器网络输出对应的游泳者特征向量。

4.如权利要求3所述的基于无监督异常检测的溺水检测方法，其特征在于，所述第一卷积自编码器的损失函数通过以下公式表述：

其中，Loss1表示损失函数，x_i表示输入图像，表示输出的重构图像，λ表示正则化的超参数，W表示第一卷积自编码器的权重参数。

5.如权利要求3所述的基于无监督异常检测的溺水检测方法，其特征在于，所述正常高斯模型的均值、协方差矩阵和概率密度函数通过以下公式表述：

其中，μ表示正常高斯模型的均值，φ(x_i)表示所述第一编码器网络的输出，m表示人体正常样本的总数量；

其中，Σ表示协方差矩阵，T表示对该向量进行转置；

其中，p(x)表示概率密度函数，n表示特征向量的维度。

6.如权利要求1所述的基于无监督异常检测的溺水检测方法，其特征在于，还包括：采用模型裁剪和量化策略对正常高斯模型进行压缩加速，以提高人体目标检测与无监督异常检测效率。

7.一种基于无监督异常检测的溺水检测装置，其特征在于，包括：

采集模块，所述采集模块用于获取水下视频流；

图像增强模块，所述图像增强模块用于对所述水下视频流进行图像增强；

目标检测模块，所述目标检测模块用于将图像增强后的水下视频流输入到预先训练好的轻量级目标检测模型，以通过所述轻量级目标检测模型判断所述水下视频流中游泳者的人体姿态是否为垂直状态；

无监督异常检测模块，所述无监督异常检测模块用于在判断结果为是时，采用深度高斯异常分类法计算该水下视频流对应的游泳者特征向量与正常高斯模型的马氏距离，并将该马氏距离作为异常得分；

判断模块，所述判断模块用于判断所述异常得分是否大于预设得分阈值，并在判断结果为是时，认为该水下视频流对应的游泳者发生溺水，并生成报警信息；

通信模块，所述通信模块用于将所述报警信息发送给相关管理人员。