摘要
本发明公开一种基于部分语义的水下遮挡目标检测方法,包括如下步骤:获取水下目标图像数据以及图像的手动标注数据,构建数据集,所述手动标注数据包括部件类别和部件框坐标、整体类别和整体框坐标;将数据集输入基于YOLO11s的检测模型中进行端到端的迭代训练,获得训练好的检测模型;所述检测模型包括有主干网络、颈部网络和引入SEM选择性增强模块的头部网络;基于自适应NMS算法进行决策优化后,输出最终检测结果;实时采集待识别的水下图像并输入训练好的检测模型中,获取水下目标的预测框与类别信息。本发明通过识别目标的部分语义特征来增强对遮挡目标的检测能力。
技术关键词
融合特征
分支
线性单元
网络
数据
坐标
多尺度特征
图像
语义特征
算法
决策
模块
元素
动态
密度
通道