基于无人船视觉的水域人员类别识别算法

作者:程亮; 吴兴辉; 江云华; 苏雄; 吴佳晓; 周辉; 丁美有; 何赟泽*
来源:电子测量与仪器学报, 2022, 36(08): 43-51.
DOI:10.13382/j.jemi.B2205344

摘要

针对水域环境下人员识别,提出了一种基于水面无人船(unmanned surface ship, USV)视觉传感器的水域人员类别识别算法。依照数据采集与模型更新流程,将采集到的视频数据进行数据清洗与标记后,创建人员类别数据集39 959张图片,7个类别;实践了基于深度学习方法下主流目标检测网络YOLO v5,并针对水域环境场景特点,提出基于YOLO v5的人员类别识别算法;将人员类别识别算法部署到边缘计算平台,实现算法在无人船上的实时应用。算法在人员类别识别数据集上达到了平均精度86%,在无人船实测中实现了每秒处理38帧的人员类别识别实时性表现。

全文