摘要:基于水域监控系统智能化的发展需求,提出了一种监控水域环境下人员识别算法。 在水域场景数据采集、数据清洗与 标记后,自主构建了一套监控水域场景下的人员类别数据集 YZ-Water4,共 8 092 张图片和 24 011 个标签。 基于目标检测算法 YOLO v7 的性能基础,针对水域场景特点,提出了适用于水域环境的目标检测算法 YOLO-WA( you only look once-water area)。 首先,使用更适合视觉任务的 FReLU 激活函数取代 YOLO v7 算法中激活函数;其次将注意力机制融合到算法网络骨架中,提升 算法的特征提取能力;最后,选择 SIOU 损失函数替换 YOLO v7 算法中的 CIOU 损失函数以优化算法训练过程。 实验结果表明, YOLO-WA 与原算法相比,在水域人员类别数据集上识别精确率由 82. 3%提升到 86. 9%,召回率由 92. 0%提升到 92. 8%,平均 精度从 88. 4%提高到 90. 6%,检测速度达到了 85 fps,满足实时运行的精度与速度要求。