摘要
本发明属于图像识别技术领域,公开了一种地质勘探场景下的小目标视觉检测方法,获取初始训练样本,初始训练样本包含地质领域中需要检测的小目标类型;构建初始深度学习网络模型;初始深度学习模型包含主干网络CSPDarknet、混合特征融合模块ASFF、地质目标特征构建模块和地质特征识别模块和一个FastestDet检测头;将初始训练样本输入到初始深度学习网络模型中进行训练,得到训练好的深度学习网络模型;将待检测图片输入训练好的深度学习网络模型VDST‑GA输出预测的目标位置和目标类型。本发明弥补了传统目标检测模型在提取微小目标所损失的特征信息,使模型更加关注小目标区域,减少背景噪声的干扰,具有准确性强、自适应性好和生产效率高的优点。
技术关键词
深度学习网络模型
注意力
特征提取模块
视觉检测方法
特征识别模块
生成式对抗网络
编码器
检测头
输出特征
解码器
通道
代表
深度学习模型
场景
矩阵
生成式网络
样本
空间金字塔