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

说明书

本发明公开了一种基于无监督异常检测的溺水检测方法及装置，其中方法包括：获取水下视频流，并对水下视频流进行图像增强；将图像增强后的水下视频流输入到预先训练好的轻量级目标检测模型，以通过轻量级目标检测模型判断水下视频流中游泳者的人体姿态是否为垂直状态；如果判断结果为是，则采用深度高斯异常分类法计算该水下视频流对应的游泳者特征向量与正常高斯模型的马氏距离，并将该马氏距离作为异常得分；判断异常得分是否大于预设得分阈值；如果是，则认为该水下视频流对应的游泳者发生溺水，并生成报警信息；能够有效提高溺水检测的精准度和实时性，并在发现溺水现象之后进行报警，进而保障游泳者的生命安全。

技术领域

本发明涉及水下图像检测技术领域，特别涉及一种基于无监督异常检测的溺水检测方法和一种基于无监督异常检测的溺水检测装置。

背景技术

随着社会的不断发展，游泳在人们的生活中也越来越普及。尤其是在夏天，游泳池是人们消暑纳凉、锻炼身体的好去处。然而，溺水事件也是频频发生。

相关技术中，在进行溺水检测时，多存在精确度地、实时性差的问题，无法在短时间内发现溺水现象并发出报警信号，从而无法保障游泳者的生命安全。

发明内容

本发明旨在至少在一定程度上解决上述技术中的技术问题之一。为此，本发明的一个目的在于提出一种基于无监督异常检测的溺水检测方法，能够有效提高溺水检测的精准度和实时性，并在发现溺水现象之后进行报警，进而保障游泳者的生命安全。

本发明的第二个目的在于提出一种基于无监督异常检测的溺水检测装置。

为达到上述目的，本发明第一方面实施例提出了一种基于无监督异常检测的溺水检测方法，包括以下步骤：获取水下视频流，并对所述水下视频流进行图像增强；将图像增强后的水下视频流输入到预先训练好的轻量级目标检测模型，以通过所述轻量级目标检测模型判断所述水下视频流中游泳者的人体姿态是否为垂直状态；如果判断结果为是，则采用深度高斯异常分类法计算该水下视频流对应的游泳者特征向量与正常高斯模型的马氏距离，并将该马氏距离作为异常得分；判断所述异常得分是否大于预设得分阈值；如果是，则认为该水下视频流对应的游泳者发生溺水，并生成报警信息。

根据本发明实施例的基于无监督异常检测的溺水检测方法，首先，获取水下视频流，并对所述水下视频流进行图像增强；接着，将图像增强后的水下视频流输入到预先训练好的轻量级目标检测模型，以通过所述轻量级目标检测模型判断所述水下视频流中游泳者的人体姿态是否为垂直状态；然后，如果判断结果为是，则采用深度高斯异常分类法计算该水下视频流对应的游泳者特征向量与正常高斯模型的马氏距离，并将该马氏距离作为异常得分；接着，判断所述异常得分是否大于预设得分阈值；然后，如果是，则认为该水下视频流对应的游泳者发生溺水，并生成报警信息；从而实现有效提高溺水检测的精准度和实时性，并在发现溺水现象之后进行报警，进而保障游泳者的生命安全。

另外，根据本发明上述实施例提出的基于无监督异常检测的溺水检测方法还可以具有如下附加的技术特征：

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

可选地，采用深度高斯异常分类法计算该水下视频流对应的游泳者特征向量与正常高斯模型的马氏距离之前，还包括：获取第一正常游泳者图像数据集，并根据所述第一正常游泳者图像数据集进行训练，以得到第一卷积自编码器，只保留第一卷积自编码器的编码器部分的网络，得到第一编码器网络；获取第二正常游泳者图像数据集，并将所述第二正常游泳者图像数据集输入到所述第一编码器网络，以通过所述第一编码器网络提取所述第二正常游泳者图像数据集中人体正常样本的特征向量；根据所述人体正常样本的特征向量构建多元高斯模型，并将所述多元高斯模型作为正常高斯模型；对所述第一编码器网络进行再次训练，以得到第二编码器网络，以便在将水下视频流输入到所述第二编码器网络中时，通过所述第二编码器网络输出对应的游泳者特征向量。