摘要
本发明涉及一种基于融合框架的无人机海洋小船只目标检测方法,包括步骤:构建目标检测模型;将扩充后的训练集输入到全局检测器中,对图像目标检测采用SimOTA进行正负样本匹配,同时采用Focal Loss函数和GIou Loss对分类损失和回归损失进行计算,通过反向传播算法更新识别全局检测器,遴选最优全局检测器;将全局检测器的输出的预测框进行NMS处理,以处理过后预测框为基础进行前景区域提取。前景块进行组装,拼装成一张图片,输入到局部目标检测器。本发明采用交叉熵损失函数和区分性对比损失函数监督模型的训练过程,得到具有更强特征学习能力的基于“全局‑局部”融合框架的海洋船只小目标检测模型,具有识别率高,可拓展性强等优点。
技术关键词
检测器
前景区域提取方法
训练集
框架
图像增强
无人机拍摄图像
组装模块
传播算法
标签
组装方法
训练分类模型
海洋船只
样本
矩形
数据
策略