摘要
本发明属于图像目标识别技术领域,具体提供基于数据增广及YOLO改进的河道目标精准检测方法,包括:采集河道水面目标图像,利用多数据增广策略对所述河道水面目标图像进行增广,获得增广图像;向YOLO网络中引入重参数化可变注意力特征提取模块以及多分支融合重构模块,结合所述增广图像,构建河道水面目标检测模型;基于所述河道水面目标检测模型对待检测河道水面目标图像进行目标检测,获得目标检测结果。本发明通过数据增广和多模块的设计,提升模型对少样本、小目标及多尺度目标的检测能力,有效抑制复杂背景干扰,更适用于河道场景。
技术关键词
融合特征
特征提取模块
精准检测方法
水面
重构模块
重构单元
多分支结构
金字塔网络
多尺度语义特征
压缩特征
图像
通道
策略
参数
生成对抗网络
数据
注意力机制