一种水体环境目标识别方法及系统

专利2025-03-03  28


本发明属于目标检测,具体涉及水体环境目标识别方法及系统。


背景技术:

1、水体环境的保护与管理是全球性的环境问题。其中,在河面、海面等水体环境中存在乱排放的固体废弃物,即为漂浮物;水面漂浮物的存在不仅影响水体的自然美观,还可能对生态系统和人类活动造成负面影响。

2、传统的水体环境目标检测方法依赖于人工巡查或简单的图像处理技术。其中,现有图像处理技术通过卷积神经网络对图像提取特征获取漂浮物识别结果,然而现有图像处理技术在准确性、效率和鲁棒性方面存在局限性。特别是在复杂的光照条件和多变的户外环境中,水面反光和漂浮物反光问题会显著降低检测的准确性。


技术实现思路

1、本发明提供了一种水体环境目标识别方法及系统,以解决水面反光和漂浮物反光显著降低检测准确性的技术问题。

2、为达到上述目的,本发明所采用的技术方案是:

3、本发明第一方面提供了一种水体环境目标识别方法,包括:

4、采集水面漂浮物监测图像,将水面漂浮物监测图像输入至预设的yolov8-omcil网络模型获得漂浮物监测结果;

5、所述yolov8-omcil网络模型的构建过程包括:

6、基于yolov8网络模型、混合网络反向残差注意力机制和跨模态注意力机制构建yolov8-omcil网络模型;

7、获取水面漂浮物目标数据,对水面漂浮物目标数据进行预处理并添加真实标签获得训练样本,通过训练样本对yolov8-omcil网络模型进行训练获得训练识别结果,根据mpdiou损失函数计算训练识别结果与训练样本之间的训练损失值,基于训练损失值对yolov8-omcil网络模型的权重参数进行优化,重复迭代yolov8-omcil网络模型的训练过程,直至达到迭代阈值输出训练后的yolov8-omcil网络模型。

8、进一步的,获取水面漂浮物目标数据,包括:

9、获取水面检测视频;按照设定帧数由水面检测视频提取水面检测图片,筛选出包含漂浮物以及图片分辨率大于设定阈值t的水面检测图片后,对水面检测图片进行标准化处理获得尺寸一致的目标图片,记为水面漂浮物目标数据。

10、进一步的,对水面漂浮物目标数据进行预处理,包括:

11、将目标图片旋转;随机将目标图片进行水平和垂直翻转,随机裁剪并缩放目标图片;随机改变目标图片的亮度、对比度、颜色饱和度;在目标图片中添加均值为0,标准差为0.1的高斯噪声。

12、进一步的,所述yolov8-omcil网络模型包括骨干网络、头部网络和任务检测网络;

13、所述骨干网络包括依次连接的第一卷积层、第二卷积层、第一特征融合层、第三卷积层、第二特征融合层、第四卷积层、第三特征融合层、第五卷积层、混合网络反向残差注意力机制模块和sppf模块;

14、所述头部网络包括依次连接的第一上采样层、第一连接层、第四特征融合层、第二上采样层、第二连接层、第五特征融合层、第一cotnet模块、第六卷积层、第三连接层、第六特征融合层、第二cotnet模块、第七卷积层、第四连接层、第七特征融合层和第三cotnet模块;

15、任务检测网络包括第一检测模块、第二检测模块和第三检测模块;

16、所述第二特征融合层的输出端与所述第二连接层;所述第三特征融合层的输出端与所述第一连接层;所述sppf模块的输出端与所述第一上采样层连接;

17、第一上采样层的输出端与所述第四连接层连接;所述第四特征融合层的输出端与所述第三连接层连接;

18、所述第六卷积层的输出端与所述第一检测模块连接;所述第二cotnet模块的输出端与所述第二检测模块连接;所述第三cotnet模块的输出端与所述第三检测模块连接。

19、进一步的,第一卷积层、第六卷积层、第七卷积层设定为conv卷积单元;第二卷积层至第五卷积层设定为odconv卷积单元;

20、所述odconv卷积单元包括依次连接的动态卷积输入层、全局平均池化层、第一全连接层和relu激活函数;所述relu激活函数连接第二全连接层、第三全连接层、第四全连接层和第五全连接层;所述第二全连接层至第四全连接层均连接单个sigmoid激活函数层;所述第五全连接层连接softmax激活函数层;所述sigmoid激活函数层和softmax激活函数层连接至n个动态卷积核;n个动态卷积核连接至第一加法器;所述第一加法器和动态卷积输入层均连接至第一乘法器;所述第一乘法器连接动态卷积输出层;

21、将特征图x输入至odconv卷积单元获得特征图y,表达公式为:

22、;

23、其中,x表示为odconv卷积单元的输入特征图,y表示为odconv卷积单元的输出特征图,表示为第i个动态卷积核的通道注意力权重,表示为第i个动态卷积核的输出通道注意力;是第i个动态卷积核的输入通道注意力;是第i个动态卷积核的空间注意力;是乘法操作;表示为第i个卷积核,i取值范围为。

24、进一步的,所述混合网络反向残差注意力模块包括依次连接的混合网络输入层、第一卷积核、第二乘法器、深度可分离卷积、第二加法器、第二卷积核、第三加法器和混合网络输出层;所述混合网络输入层连接第三加法器和注意力矩阵单元;注意力矩阵单元与第二乘法器连接;所述第二乘法器输出端连接所述第二加法器;所述第一卷积核和第二卷积核设定为1×1卷积。

25、进一步的,所述第一cotnet模块、第二cotnet模块和第三cotnet模块包括跨模态输入层;所述跨模态输入层连接第三卷积核、第四卷积核和第五连接层;第四卷积核、第五连接层、第五卷积核、第六卷积核、第三乘法器、第四加法器和跨模态输出层依次连接;第三卷积核的输出端连接第六卷积核;所述第四卷积核的输出端连接第四加法器。

26、进一步的,通过训练样本对yolov8-omcil网络模型进行训练获得训练识别结果,包括:

27、将训练样本输入至所述yolov8-omcil网络模型的骨干网络;由骨干网络中sppf模块输出特征图p1,将特征图p1输入至第一上采样层;通过第一上采样层对特征图p1进行上采样获得特征图q1;

28、由第三特征融合层输出特征图p2,通过第一连接层将特征图q1和特征图p2进行拼接后,并依次发送至第四特征融合层;由第四特征融合层输出特征图q2,通过第二上采样层对特征图q2进行上采样获得特征图q3;

29、由第二特征融合层输出特征图p3,通过第二连接层将特征图q3和特征图p3进行拼接后,并依次发送至第五特征融合层、第一cotnet模块和第六卷积层,由第六卷积层生成检测特征图u1并发送至第一检测模块;

30、通过第三连接层将检测特征图u1和特征图q2进行拼接后,并依次发送至第六特征融合层和第二cotnet模块;由第二cotnet模块输出检测特征图u2;

31、将检测特征图u2依次输入至第七卷积层和第四连接层;通过第四连接层将第七卷积层的输出特征与特征图q1进行拼接后,并依次发送至第七特征融合层和第三cotnet模块;所述第三cotnet模块输出检测特征图u3;

32、对检测特征图u1、检测特征图u2和检测特征图u3进行检测获得漂浮物识别框架,记为训练识别结果。

33、进一步的,根据mpdiou损失函数计算训练识别结果与训练样本之间的训练损失值,表达公式为:

34、;

35、;

36、;

37、其中,表示为漂浮物识别框架的左上角点坐标;表示为漂浮物识别框架的右下角点坐标;表示为真实标签中漂浮物框架的左上角点坐标;为真实标签中漂浮物框架的右下角点坐标;为预漂浮物识别框架与真实标签中漂浮物框架的左上角点之间欧几里得距离的平方;为预漂浮物识别框架与真实标签中漂浮物框架的右下角点之间欧几里得距离的平方;w和h分别表示真实标签中漂浮物框架的宽度和高度;为预漂浮物识别框架与真实标签中漂浮物框架的交并比。

38、本发明第二方面提供了一种水体环境目标识别系统,包括:

39、监测模块,用于采集水面漂浮物监测图像,将水面漂浮物监测图像输入至预设的yolov8-omcil网络模型获得漂浮物监测结果;

40、模型构建模块,用于基于yolov8网络模型、混合网络反向残差注意力机制和跨模态注意力机制构建yolov8-omcil网络模型;

41、获取模块,用于获取水面漂浮物目标数据,对水面漂浮物目标数据进行预处理并添加真实标签获得训练样本,

42、训练模块,用于通过训练样本对yolov8-omcil网络模型进行训练获得训练识别结果;根据mpdiou损失函数计算训练识别结果与训练样本之间的训练损失值,基于训练损失值对yolov8-omcil网络模型的权重参数进行优化,重复迭代yolov8-omcil网络模型的训练过程,直至达到迭代阈值输出训练后的yolov8-omcil网络模型。

43、与现有技术相比,本发明的有益效果:

44、本发明采集水面漂浮物监测图像,将水面漂浮物监测图像输入至预设的yolov8-omcil网络模型获得漂浮物监测结果;能够适应不同的环境条件和目标特性,提高了水体环境目标检测的实时性和可靠性。

45、本发明基于yolov8网络模型、混合网络反向残差注意力机制和跨模态注意力机制构建yolov8-omcil网络模型;通过改进的yolov8算法,增强了模型对不同光照条件下水体环境目标特征的提取能力,优化了模型的收敛过程,提高了在不同天气下对水体环境目标识别的鲁棒性。

46、本发明根据mpdiou损失函数计算训练识别结果与训练样本之间的训练损失值,基于训练损失值对yolov8-omcil网络模型的权重参数进行优化,使用mpdiou损失函数替代传统损失函数,通过计算预测框和真实框的特定点之间的距离,增加了回归的准确性,优化了模型对远距离难识别易混淆小目标的定位精度。


技术特征:

1.一种水体环境目标识别方法,其特征在于,包括:

2.根据权利要求1所述的水体环境目标识别方法,其特征在于,获取水面漂浮物目标数据,包括:

3.根据权利要求1所述的水体环境目标识别方法,其特征在于,对水面漂浮物目标数据进行预处理,包括:

4.根据权利要求1所述的水体环境目标识别方法,其特征在于,所述yolov8-omcil网络模型包括骨干网络、头部网络和任务检测网络;

5.根据权利要求4所述的水体环境目标识别方法,其特征在于,第一卷积层、第六卷积层、第七卷积层设定为conv卷积单元;第二卷积层至第五卷积层设定为odconv卷积单元;

6.根据权利要求4所述的水体环境目标识别方法,其特征在于,所述混合网络反向残差注意力模块包括依次连接的混合网络输入层、第一卷积核、第二乘法器、深度可分离卷积、第二加法器、第二卷积核、第三加法器和混合网络输出层;所述混合网络输入层连接第三加法器和注意力矩阵单元;注意力矩阵单元与第二乘法器连接;所述第二乘法器输出端连接所述第二加法器;所述第一卷积核和第二卷积核设定为1×1卷积。

7.根据权利要求4所述的水体环境目标识别方法,其特征在于,所述第一cotnet模块、第二cotnet模块和第三cotnet模块包括跨模态输入层;所述跨模态输入层连接第三卷积核、第四卷积核和第五连接层;第四卷积核、第五连接层、第五卷积核、第六卷积核、第三乘法器、第四加法器和跨模态输出层依次连接;第三卷积核的输出端连接第六卷积核;所述第四卷积核的输出端连接第四加法器。

8.根据权利要求4所述的水体环境目标识别方法,其特征在于,通过训练样本对yolov8-omcil网络模型进行训练获得训练识别结果,包括:

9.根据权利要求1所述的水体环境目标识别方法,其特征在于,根据mpdiou损失函数计算训练识别结果与训练样本之间的训练损失值,表达公式为:

10.一种水体环境目标识别系统,其特征在于,包括:


技术总结
本发明公开了一种水体环境目标识别方法及系统,方法包括:采集水面漂浮物监测图像,基于yolov8网络模型、混合网络反向残差注意力机制和跨模态注意力机制构建yolov8‑OMCIL网络模型,记为漂浮物识别模型;通过训练样本对漂浮物识别模型进行训练获得训练识别结果,计算训练识别结果与训练样本之间的训练损失值,基于训练损失值对漂浮物识别模型的权重参数进行优化,输出训练后的漂浮物识别模型;将水面漂浮物监测图像输入至预设的漂浮物识别模型获得漂浮物监测结果;本发明增强了对不同光照条件下水体环境目标特征的提取能力,优化了模型的收敛过程,提高了在不同天气下对水体环境目标识别的鲁棒性。

技术研发人员:毕鹏飞,胡志远,周楠翔,张研博
受保护的技术使用者:南京信息工程大学
技术研发日:
技术公布日:2024/11/11
转载请注明原文地址: https://tieba.8miu.com/read-11883.html

最新回复(0)