[发明专利]泳池防溺水人体目标检测方法、装置、计算机设备及存储介质

权利要求书

1.泳池防溺水人体目标检测方法，其特征在于，包括：

获取位于泳池水面上方的多个相机所拍摄的图像，以得到原始图像；

将所述原始图像输入至目标检测模型内进行泳池防溺水人体目标检测，以得到检测结果；

输出所述检测结果；

其中，所述目标检测模型是通过若干个带有人体目标所在位置标签的图像作为样本集训练卷积神经网络所得的；

所述目标检测模型是通过若干个带有人体目标所在位置标签的图像作为样本集训练卷积神经网络所得的，包括：

构建带有人体目标所在位置标签的图像作为样本集；

对所述样本集进行划分，以得到训练集、验证集以及测试集；

对所述训练集、验证集以及测试集进行增强处理，以得到处理结果；

构建Yolov5网络，并在Yolov5网络添加DLA-34网络、Semantic Self-Attention机制以及Anchor-free网络，以得到初始网络；

利用所述处理结果中的训练集、验证集对所述初始网络进行训练和验证，并计算训练过程的损失值；

当损失值维持不变时，利用所述处理结果中的测试集对所述初始网络进行测试，以将训练后的初始网络作为目标检测模型。

2.根据权利要求1所述的泳池防溺水人体目标检测方法，其特征在于，所述构建Yolov5网络，并在Yolov5网络添加DLA-34网络、Semantic Self-Attention机制以及Anchor-free网络，以得到初始网络，包括：

构建Yolov5网络；

在所述Yolov5网络添加DLA-34网络作为主干网络，进行提取特征，以得到第一网络；

在所述第一网络添加Semantic Self-Attention机制，以得到第二网络；

在第二网络的目标回归框网络中使用Anchor-free网络，以得到初始网络。

3.根据权利要求1所述的泳池防溺水人体目标检测方法，其特征在于，所述检测结果包括泳池人体目标所在的位置。

4.泳池防溺水人体目标检测装置，其特征在于，包括：

图像获取单元，用于获取位于泳池水面上方的多个相机所拍摄的图像，以得到原始图像；

目标检测单元，用于将所述原始图像输入至目标检测模型内进行泳池防溺水人体目标检测，以得到检测结果；

输出单元，用于输出所述检测结果；

所述目标检测模型是通过若干个带有人体目标所在位置标签的图像作为样本集训练卷积神经网络所得的，包括：

构建带有人体目标所在位置标签的图像作为样本集；

对所述样本集进行划分，以得到训练集、验证集以及测试集；

对所述训练集、验证集以及测试集进行增强处理，以得到处理结果；

构建Yolov5网络，并在Yolov5网络添加DLA-34网络、Semantic Self-Attention机制以及Anchor-free网络，以得到初始网络；

利用所述处理结果中的训练集、验证集对所述初始网络进行训练和验证，并计算训练过程的损失值；

当损失值维持不变时，利用所述处理结果中的测试集对所述初始网络进行测试，以将训练后的初始网络作为目标检测模型。

5.根据权利要求4所述的泳池防溺水人体目标检测装置，其特征在于，还包括模型生成单元；

所述模型生成单元，用于通过若干个带有人体目标所在位置标签的图像作为样本集训练卷积神经网络，以得到目标检测模型。